Home
WorkBlogTalksPapersAbout
Nikita Kozodoi

Nikita Kozodoi, PhD

Senior Scientist at AWS Forward Deployed Engineering. I publish applied AI research and embed with engineering teams to build custom AI systems, from experimentation to production.

8+ years in AI/ML · Focus on LLMs and agentic AI

CV
29
Blog Posts
40
Public Talks
19
Papers
659
Paper Citations
405
GitHub Stars
18
Kaggle Medals

What's new

Blog

SWE-InfraBench + Kiro: What Happens When a Coding Agent Tackles IaC?

Builder Center

Language models solved only 34% of SWE-InfraBench's infrastructure-as-code tasks in one attempt. We ran Kiro over all 100 and found that a full agent loop with Sonnet 5 reaches 82%, with partially correct answers converging once the agent can re-run the tests itself.

View all 29 blog posts
Talk

Test-Time Augmentation for LLMs: When Input Diversity Beats Output Diversity at Matched Compute

COLM 2026 Workshop on Efficient Reasoning · San Francisco

Test-time scaling improves LLM accuracy but multiplies inference cost, making the accuracy gained per unit of compute the metric that matters in deployment. Self-consistency is one of the established approaches, which spends this budget entirely on the output side by sampling repeated reasoning paths. We study Test-Time Augmentation (TTA), which extends self-consistency by also perturbing the input, aggregating predictions across transformed versions of the input, and ask whether input-side diversity converts compute into accuracy more efficiently than output-side diversity. We perform a systematic, matched-compute comparison: we evaluate three simple input-side strategies (semantic rephrasing, lexical perturbations, and visual transformations) across six datasets covering general and multilingual knowledge, mathematical reasoning, multi-modal question answering, and sentiment classification, against chain-of-thought prompting and self-consistency. Semantic rephrasing delivers consistent and statistically significant accuracy gains while Pareto-dominating self-consistency on cost-effectiveness, delivering roughly 1.8X more accuracy per dollar and outperforming it on five of six tasks. We further analyze the number of augmentations, multi-modal strategies, and base model scaling, finding that TTA is most cost-effective for mid-tier models where a stronger model is unavailable or too expensive. Our findings indicate that for current mid-tier LLMs, varying the input converts inference compute into accuracy more efficiently than varying the reasoning path alone.

View all 40 talks
Paper

Evolving LLM-Generated Features for Interpretable Classification

NeurIPS 2026 Workshop on Trust in AI Evaluation

Large language models (LLMs) are increasingly used as classifiers, yet they operate as opaque systems whose decisions are difficult to interpret, which complicates their use in regulated domains such as credit scoring or medical diagnosis. We propose an evolutionary framework that iteratively discovers natural language feature definitions (rubrics) for interpretable classification. An LLM generates candidate binary features, evaluates each sample against them, and the resulting vectors can be used to train a transparent classifier such as logistic regression. The feature set evolves over multiple iterations guided by classification errors, per-class activation rates, and feature ablation scores. We evaluate across three benchmarks, including a credit risk dataset representative of regulated domains, comparing single-shot LLM rubrics, evolved rubrics, and direct zero-shot LLM classification. Evolved features improve over single-shot rubrics by +2.9 pp on average and outperform zero-shot LLM classification on two of three tasks, while providing fully auditable decision logic. On the credit risk task, the zero-shot LLM performs at chance (50.7%) with a strong bias toward a single class, whereas evolved features achieve balanced, interpretable predictions. Crucially, this failure is invisible in aggregate accuracy and surfaces only under per-class auditing. Analysis reveals that evolution is most effective when label boundaries cannot be inferred from category names alone or when the LLM lacks reliable domain-specific reasoning.

View all 19 papers

Latest on LinkedIn

Follow on LinkedIn
11 Oct 2026

Excited to present 3 papers on LLM optimization, scaling and merging at COLM in San Francisco! 🇺🇸 Always happy to chat with fellow scientists at the Amazon booth and exchange ideas on where the field is going, and how to keep up with the speed of research. Kudos to my co-authors from the AWS FDE team: Jack Butler, Zainab A., Egor Krasheninnikov, Sergey I., Natalia Tarasova

8251
2 Oct 2026

Great energy at AWS Summit Dubai! It was inspiring to meet so many builders and showcase our team’s work to customers across the region.

5720
24 Sept 2026

Meet us at AWS Summit Dubai next week! 🇦🇪 Our team will demo custom solutions we built for real business use cases, including voice and video AI, deep research agents, and physical AI. If you want to learn more about AWS Forward Deployed Engineering, and chat about agentic AI and model customization - find us at the AI/ML zone! 📅 September 30, Dubai World Trade Centre 🔗 Register: https://lnkd.in/ehjXFrW3

2411

Selected work

All 8 case studies
AUMOVIO

Multi-agent defect detection for automotive software

Agentic code scanning system that found 30+ high-priority defects in a production automotive codebase.

Toyota Motor Europe

Recovering business logic from a mainframe

Agentic pipelines that recovered the business logic buried in 1.3 million lines of mainframe code, now moving to production rollout.

Lounge by Zalando

Localizing marketing content across 17 European markets

Test-time self-reflection for content localization across 17 European markets, tuned for quality, cost and latency.

Get in touch

Open to collaboration, speaking invitations, and conversations about applied AI.

Contact

Pages

  • Work
  • Blog
  • Talks
  • Papers
  • About

Social

  • LinkedIn
  • GitHub
  • Google Scholar
  • X
  • Instagram

Contact

  • n.kozodoi@icloud.com
  • Buy me a coffee
  • Download CV
  • RSS feed
  • Impressum

© 2026 Nikita Kozodoi. All opinions are my own.