Research
* denotes equal contributions. Also see my Google Scholar.
Optimizing Analytic Constants via AI-Guided Lean Proof Refinement
Rahul Saha*, Alan Li*, Anton Xue, Adam Klivans, Pravesh Kothari, Raghu Meka, and Swarat Chaudhuri
NeurIPS 2026
Multi-Target Few-Shot Prompting with Diffusion Language Models
Anton Xue, Alan Li, and Rahul Saha
arXiv 2026
PIG: Parallel Inconsistency Guidance for Accelerating Diffusion Language Models
Quentin Fruytier, Anton Xue, Sanjay Shakkottai, and Sujay Sanghavi
arXiv 2026
LESSER: Post-Training Data Selection with Output-Layer Gradients
Lyuxin David Zhang, Eric Wong, Surbhi Goel, and Anton Xue
arXiv 2026
dQwen3.5: Hybrid-Attention Diffusion Language Models
Anton Xue, Litu Rout, Aditya Akella, Adam Klivans, Sujay Sanghavi, and Sanjay Shakkottai
arXiv 2026
New Lower and Upper Bounds for the Grothendieck Constant
Rahul Saha*, Alan Li*, Anton Xue, Swarat Chaudhuri, Adam Klivans, Pravesh Kothari, and Raghu Meka
arXiv 2026
Long-Horizon AI Research for Grothendieck Constant: A Case Study in Human-AI Mathematical Collaboration
Alan Li*, Rahul Saha*, Anton Xue, Swarat Chaudhuri, Adam Klivans, Pravesh Kothari, and Raghu Meka
arXiv 2026
The Grothendieck Constant is Less Than π/(2 log(1+√2)) − 10−5
Alan Li*, Rahul Saha*, Anton Xue, Swarat Chaudhuri, Adam Klivans, Pravesh Kothari, and Raghu Meka
arXiv 2026
Missingness Bias Calibration in Feature Attribution Explanations
Shailesh Sridhar, Anton Xue, and Eric Wong
ICLR 2026
AnCoder: Anchored Code Generation via Discrete Diffusion Models
Anton Xue*, Litu Rout*, Constantine Caramanis, and Sanjay Shakkottai
arXiv 2026
The Transformer Cookbook
Andy Yang, Christopher Watson, Anton Xue, Satwik Bhattamishra, Jose Llarena, William Merrill, Emile Dos Santos Ferreira, Anej Svete, and David Chiang
TMLR 2026
Probabilistic Stability Guarantees for Feature Attributions
Helen Jin*, Anton Xue*, Weiqiu You, Surbhi Goel, and Eric Wong
NeurIPS 2025 · code · blog
T-FIX: Text-Based Explanations with Features Interpretable to eXperts
Shreya Havaldar*, Weiqiu You*, Chaehyeon Kim, Anton Xue, Helen Jin, Marco Gatti, Bhuvnesh Jain, Helen Qu, Amin Madani, Daniel Hashimoto, Gary Weissman, Rajat Deo, Sameed Khatana, Lyle Ungar, and Eric Wong
arXiv 2025
Probabilistic Soundness Guarantees in LLM Reasoning Chains
Weiqiu You, Anton Xue, Shreya Havaldar, Delip Rao, Helen Jin, Chris Callison-Burch, and Eric Wong
EMNLP 2025 · code · blog
The FIX Benchmark: Extracting Features Interpretable to eXperts
Helen Jin*, Shreya Havaldar*, Chaehyeon Kim*, Anton Xue*, Weiqiu You*, Helen Qu, Marco Gatti, Daniel Hashimoto, Bhuvnesh Jain, Amin Madani, Masao Sako, Lyle Ungar, and Eric Wong
DMLR 2025 · code
On The Concurrence of Layer-wise Preconditioning Methods and Provable Feature Learning
Thomas Zhang*, Behrad Moniri*, Ansh Nagwekar, Faraz Rahman, Anton Xue, Hamed Hassani, and Nikolai Matni
ICML 2025
Logicbreaks: A Framework for Understanding Subversion of Rule-based Inference
Anton Xue*, Avishree Khare*, Rajeev Alur, Surbhi Goel, and Eric Wong
ICLR 2025 · code · blog
AR-Pro: Counterfactual Explanations for Anomaly Repair with Formal Properties
Xiayan Ji*, Anton Xue*, Eric Wong, Oleg Sokolsky, and Insup Lee
NeurIPS 2024
TYGR: Type Inference on Stripped Binaries using Graph Neural Networks
Chang Zhu, Ziyang Li, Anton Xue, Ati Priya Bajaj, William Gibbs, Yibo Liu, Rajeev Alur, Tiffany Bao, Hanjun Dai, Adam Doupé, Mayur Naik, Yan Shoshitaishvili, Ruoyu Wang, and Aravind Machiry
USENIX Security 2024
Chordal sparsity for SDP-based neural network verification
Anton Xue, Lars Lindemann, and Rajeev Alur
Automatica 2024 · code
Stability Guarantees for Feature Attributions with Multiplicative Smoothing
Anton Xue, Rajeev Alur, and Eric Wong
NeurIPS 2023 · code · blog
Chordal Sparsity for Lipschitz Constant Estimation of Deep Neural Networks
Anton Xue, Lars Lindemann, Alexander Robey, Hamed Hassani, George Pappas, and Rajeev Alur
CDC 2022 · code
Safe Reinforcement Learning via Statistical Model Predictive Shielding
Osbert Bastani, Shuo Li, and Anton Xue
RSS 2021
Data-Driven System Level Synthesis
Anton Xue and Nikolai Matni
L4DC 2021 · code
Synchronization Schemas
Rajeev Alur, Phillip Hilliard, Zachary Ives, Konstantinos Kallas, Konstantinos Mamouras, Filip Niksic, Caleb Stanford, Val Tannen, and Anton Xue
Principles of Database Systems 2021
A Self-certifying Compilation Framework for WebAssembly
Kedar Namjoshi and Anton Xue
VMCAI 2021 · code
G2Q: Haskell constraint solving
William Hallahan, Anton Xue, and Ruzica Piskac
Haskell Symposium 2019
Lazy counterfactual symbolic execution
William Hallahan, Anton Xue, Maxwell Troy Bland, Ranjit Jhala, and Ruzica Piskac
PLDI 2019 · code · video