Ax Aneesh Pappu, Batu El, Hancheng Cao, Carmelo di Nolfo, Yanchao Sun, Meng Cao, James Zou 6/1/2026

Multi-Agent Teams Hold Experts Back

Study shows self-organizing multi-agent LLM teams underperform expert models; examines coordination emergence in autonomous collaborators.

Ax Fabian Fumagalli, Landon Butler, Justin Singh Kang, Kannan Ramchandran, R. Teal Witter 6/1/2026

An Odd Estimator for Shapley Values

Novel estimator for computing Shapley values with theoretical analysis of paired sampling heuristic for attribution.

Ax Francesco De Santis, Gabriele Ciravegna, Giovanni De Felice, Arianna Casanova, Francesco Giannini, Michelangelo Diligenti, Johannes Schneider, Danilo Giordano, Mateo Espinosa Zarlenga, Pietro Barbiero 6/1/2026

Mixture of Concept Bottleneck Experts

Mixture of Concept Bottleneck Experts framework improves interpretability and accuracy of concept-based models with multiple task predictors.

Ax Qinglin Zhu, Tianyu Chen, Shuai Lu, Lei Ji, Runcong Zhao, Murong Ma, Xiangxiang Dai, Yulan He, Lin Gui, Peng cheng, Yeyun Gong 6/1/2026

Pull Requests as a Training Signal for Repo-Level Code Editing

Clean-PR training paradigm leverages GitHub pull requests to improve repository-level code editing models for multi-file modifications.

Ax Ziyun Li, Huancheng Hu, Soon Hoe Lim, Xuyu Li, Fei Gao, Enmao Diao, Zezhen Ding, Michalis Vazirgiannis, Henrik Bostrom 6/1/2026

A Kinetic Energy Perspective of Flow Matching

Physics-inspired diagnostic for flow-based generative models measuring kinetic effort along ODE trajectories.

Ax Jinwoo Kim, S\'ekou-Oumar Kaba, Jiyun Park, Seunghoon Hong, Siamak Ravanbakhsh 6/1/2026

Inverting Data Transformations via Diffusion Sampling

Probabilistic approach to inverting unknown group transformations using diffusion sampling for transformation recovery on Lie groups.

Ax Iv\'an Arcuschin, David Chanin, Adri\`a Garriga-Alonso, Oana-Maria Camburu 6/1/2026

Biases in the Blind Spot: Detecting What LLMs Fail to Mention

Proposes automated pipeline for detecting unverbalized biases in LLM chain-of-thought reasoning without requiring predefined categories or hand-crafted datasets.

Ax Tessa Han, Sebastian Bordt, Hanlin Zhang, Sham Kakade 6/1/2026

Weight Decay Improves Language Model Plasticity

Studies how weight decay during pretraining improves LLM plasticity and downstream adaptability beyond validation loss optimization.

Ax Yuxiang Guo, Zhuoran Du, Nan Tang, Kezheng Tang, Congcong Ge, Yunjun Gao 6/1/2026

DTBench: A Synthetic Benchmark for Document-to-Table Extraction

DTBench introduces a synthetic benchmark for evaluating LLM performance on document-to-table extraction tasks requiring complex reasoning and structured output generation.

Ax Kiho Park, Todd Nief, Yo Joong Choe, Victor Veitch 6/1/2026

The Information Geometry of Softmax: Probing and Steering

Studies information geometry of softmax representations in AI systems, focusing on how models encode semantic structure into representation spaces for behavior production.

Ax Yufei Li, Yisen Gao, Jiaxuan Xiong, Jiaxin Bai, Shijie Zhong, Haoyu Huang, Zhongwei Xie, Hong Ting Tsang, Yangqiu Song 6/1/2026

NGDBench: Towards Neural Graph Data Management

NGDBench proposes next-generation neural data management systems supporting heterogeneous evolving data with implicit reasoning.

Ax Charles Ye, Jasmine Cui, Dylan Hadfield-Menell 6/1/2026

Prompt Injection as Role Confusion

Research on prompt injection attacks as role confusion, showing LLMs identify text source by style rather than role labels, with measurement techniques.

Ax Rajinder Sandhu, Di Mu, Cheng Chang, Md Shahriar Tasjid, Himanshu Rai, Maksims Volkovs, Ga Wu 6/1/2026

Aligning Dense Retrievers with LLM Utility via Distillation

Utility-Aligned Embeddings framework improving dense retrieval for RAG by distilling LLM utility signals without expensive re-ranking.

Ax Bowen Zheng, Weijian Luo, Guang Yang, Colin Zhang, Tianyang Hu 6/1/2026

Autoregressive Visual Generation Needs a Prologue

Prologue approach bridging reconstruction-generation gap in autoregressive image generation by using separate tokens for each objective.