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
20
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

Generation and Verification: Human-Agent Collaboration for Safety-Critical Automotive Software Engineering

EMNLP 2026 Workshop on Research on Agent Language Models

Research on coding agents mainly focuses on generation quality. Yet, safety-critical automotive software makes two additional demands. It prioritizes verification, since every change is reviewed, traced to a requirement, and checked against industry standards such as MISRA and AUTOSAR before it can be committed. It is also a narrow domain, written against layered architectures, embedded resource constraints, and organization-specific interface conventions that are scarce in the data general-purpose LLMs are trained on. We present a coding assistant for automotive C/C++ that addresses both demands. First, it exposes three tools to the user over the Model Context Protocol (generate, edit, and verify), allowing developers to run verify on any code on demand rather than relying on the agent to check its own output. Second, rather than using a general-purpose model, we customize the underlying LLM for the automotive domain through prompt engineering and fine-tuning. We construct a dataset from automotive production code of AUMOVIO, where each function is paired with the requirement texts and Unified Modeling Language (UML) behavior it must satisfy. In a two-week pilot on active production projects, developers report verify as the most-used tool and confirm using it on their own code and not only on generated output. Comparing model customization strategies, 75% prefer the faster prompt-engineered configuration. MISRA- and AUTOSAR-inspired pattern checkers indicate that prompt engineering achieves a lower number of violations on C, while fine-tuning shows better conformance with architecture patterns for C++.

View all 20 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.