Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Thomas Kwa

dblp:267/9816 · DBLP profile ↗
← Back
4ranked-venue papers
2as first author
4since 2021 · last 2025
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 4 · 2 first-author · 4 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
4 papers
Trustworthy machine learning · 51% Reinforcement learning · 40% Deep learning architectures and training · 9%
Interdisciplinary, comprehensive, and emerging computing
1 paper
Computing education · 100%
Software engineering, system software, and programming languages
1 paper
Program verification · 100%

Topics — the 9 heaviest of 11, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Trustworthy machine learning
interpretability
1.522024
InterpBench: Semi-Synthetic Transformers for Evaluating Mechanistic Interpretability Techniques · NeurIPS 2024
Compact Proofs of Model Performance via Mechanistic Interpretability · NeurIPS 2024
Machine learning › Trustworthy machine learning › interpretability
mechanistic interpretability
1.522024
InterpBench: Semi-Synthetic Transformers for Evaluating Mechanistic Interpretability Techniques · NeurIPS 2024
Compact Proofs of Model Performance via Mechanistic Interpretability · NeurIPS 2024
Machine learning › Trustworthy machine learning
AI safety
0.912025
Measuring AI Ability to Complete Long Software Tasks · NeurIPS 2025
Machine learning › Reinforcement learning › regularization for reinforcement learning
KL regularization
0.812024
Catastrophic Goodhart: regularizing RLHF with KL divergence does not mitigate heavy-tailed reward misspecification · NeurIPS 2024
Machine learning › Reinforcement learning
reinforcement learning from human feedback
0.812024
Catastrophic Goodhart: regularizing RLHF with KL divergence does not mitigate heavy-tailed reward misspecification · NeurIPS 2024
Machine learning › Reinforcement learning › reward design
reward misspecification
0.812024
Catastrophic Goodhart: regularizing RLHF with KL divergence does not mitigate heavy-tailed reward misspecification · NeurIPS 2024
Machine learning › Reinforcement learning › reinforcement learning from human feedback › learning from human feedback
RLHF
0.812024
Catastrophic Goodhart: regularizing RLHF with KL divergence does not mitigate heavy-tailed reward misspecification · NeurIPS 2024
Machine learning › Deep learning architectures and training
transformer
0.522024
InterpBench: Semi-Synthetic Transformers for Evaluating Mechanistic Interpretability Techniques · NeurIPS 2024
Compact Proofs of Model Performance via Mechanistic Interpretability · NeurIPS 2024
Machine learning › Deep learning architectures and training › transformer › efficient transformer
sparse transformer
0.212024
InterpBench: Semi-Synthetic Transformers for Evaluating Mechanistic Interpretability Techniques · NeurIPS 2024

Methods — techniques the papers use, named apart from their topics

human task timing · 1.7benchmark extrapolation · 1.7mechanistic interpretability · 1.5computer-assisted proof · 1.5interchange intervention training · 0.8discrete optimization for tail measurement · 0.8circuit discovery · 0.8
YearPublicationVenuePosition
2025 Measuring AI Ability to Complete Long Software Tasks
abstract
Despite rapid progress on AI benchmarks, the real-world meaning of benchmark performance remains unclear. To quantify the capabilities of AI systems in terms of human capabilities, we propose a new metric: 50%-task-completion time horizon. This is the time humans typically take to complete tasks that AI models can complete with 50% success rate. We first timed humans with relevant domain expertise on a combination of RE-Bench, HCAST, and 66 novel shorter tasks. On these tasks, current frontier AI models such as o3 have a 50% time horizon of around 110 minutes. Furthermore, frontier AI time horizon has been doubling approximately every seven months since 2019, though the trend may have accelerated since 2024. The increase in AI models’ time horizons seems to be primarily driven by greater reliability and ability to adapt to mistakes, combined with better logical reasoning and tool use capabilities. We discuss the limitations of our results—including their degree of external validity—and the implications of increased autonomy for dangerous capabilities. If these results generalize to real-world software tasks, extrapolation of this trend predicts that within 5 years, AI systems will be capable of automating many software tasks that currently take humans a month.
Thomas Kwa, Ben West, Joel Becker, Amy Deng, Katharyn Garcia, Max Hasin, Sami Jawhar, Megan Kinniment, Nate Rush, Sydney von Arx, Ryan Bloom, Thomas Broadley, Haoxing Du, Brian Goodrich, Nikola Jurkovic, Luke Harold Miles, Seraphina Nix, Tao Roa Lin, Neev Parikh, David Rein, Lucas Jun Koba Sato, Hjalmar Wijk, Daniel M. Ziegler, Elizabeth Barnes, Lawrence Chan
NeurIPS1
2024 Compact Proofs of Model Performance via Mechanistic Interpretability
abstract
We propose using mechanistic interpretability -- techniques for reverse engineering model weights into human-interpretable algorithms -- to derive and compactly prove formal guarantees on model performance. We prototype this approach by formally proving accuracy lower bounds for a small transformer trained on Max-of-$K$, validating proof transferability across 151 random seeds and four values of $K$. We create 102 different computer-assisted proof strategies and assess their length and tightness of bound on each of our models. Using quantitative metrics, we find that shorter proofs seem to require and provide more mechanistic understanding. Moreover, we find that more faithful mechanistic understanding leads to tighter performance bounds. We confirm these connections by qualitatively examining a subset of our proofs. Finally, we identify compounding structureless errors as a key challenge for using mechanistic interpretability to generate compact proofs on model performance.
Jason Gross, Rajashree Agrawal, Thomas Kwa, Euan Ong, Chun Hei Yip, Alex Gibson, Soufiane Noubir, Lawrence Chan
NeurIPS3
2024 InterpBench: Semi-Synthetic Transformers for Evaluating Mechanistic Interpretability Techniques
abstract
Mechanistic interpretability methods aim to identify the algorithm a neural network implements, but it is difficult to validate such methods when the true algorithm is unknown. This work presents InterpBench, a collection of semi-synthetic yet realistic transformers with known circuits for evaluating these techniques. We train simple neural networks using a stricter version of Interchange Intervention Training (IIT) which we call Strict IIT (SIIT). Like the original, SIIT trains neural networks by aligning their internal computation with a desired high-level causal model, but it also prevents non-circuit nodes from affecting the model's output. We evaluate SIIT on sparse transformers produced by the Tracr tool and find that SIIT models maintain Tracr's original circuit while being more realistic. SIIT can also train transformers with larger circuits, like Indirect Object Identification (IOI). Finally, we use our benchmark to evaluate existing circuit discovery techniques.
Rohan Gupta, Iván Arcuschin, Thomas Kwa, Adrià Garriga-Alonso
NeurIPS3
2024 Catastrophic Goodhart: regularizing RLHF with KL divergence does not mitigate heavy-tailed reward misspecification
abstract
When applying reinforcement learning from human feedback (RLHF), the reward is learned from data and, therefore, always has some error. It is common to mitigate this by regularizing the policy with KL divergence from a base model, with the hope that balancing reward with regularization will achieve desirable outcomes despite this reward misspecification. We show that when the reward function has light-tailed error, optimal policies under less restrictive KL penalties achieve arbitrarily high utility. However, if error is heavy-tailed, some policies obtain arbitrarily high reward despite achieving no more utility than the base model--a phenomenon we call catastrophic Goodhart. We adapt a discrete optimization method to measure the tails of reward models, finding that they are consistent with light-tailed error. However, the pervasiveness of heavy-tailed distributions in many real-world applications indicates that future sources of RL reward could have heavy-tailed error, increasing the likelihood of reward hacking even with KL regularization.
Thomas Kwa, Drake Thomas, Adrià Garriga-Alonso
NeurIPS1