Ax Xiaoxuan Liang, Saeid Naderiparizi, Yunpeng Liu, Berend Zwartsenberg, Frank Wood 3/10/2026

Improved Constrained Generation by Bridging Pretrained Generative Models

Constrained generation framework for diffusion models handling complex feasible regions in robotics and autonomous driving applications.

Ax Jianyuan Zhong, Kaibo Wang, Ding Ding, Zijin Feng, Haoli Bai, Yang Xiang, Jiacheng Sun, Qiang Xu 3/10/2026

Stabilizing Reinforcement Learning for Diffusion Language Models

Method to stabilize Group Relative Policy Optimization (GRPO) for diffusion language models by addressing reward collapse issues in post-training.

Ax Minjae Kang, Jaehyung Kim 3/10/2026

Enhancing Instruction Following of LLMs via Activation Steering with Dynamic Rejection

DIRECTER: Dynamic rejection steering technique for LLMs to improve instruction following while avoiding oversteering that degrades output quality.

Ax Gregor Baer 3/10/2026

xaitimesynth: A Python Package for Evaluating Attribution Methods for Time Series with Synthetic Ground Truth

xaitimesynth: Reusable Python package for evaluating time series attribution methods using synthetic ground truth data with class-discriminating features.

Ax Mai Pham, Vikrant Vaze, Peter Chin 3/10/2026

Optimistic Policy Regularization

OPR: Lightweight mechanism preventing premature policy convergence in deep RL by maintaining buffer of high-performing episodes during optimization.

Ax Irene Wang, Vishnu Varma Venkata, Arvind Krishnamurthy, Divya Mahajan 3/10/2026

NEST: Network- and Memory-Aware Device Placement For Distributed Deep Learning

NEST: Device placement optimization for distributed deep learning that jointly considers parallelism, memory, and network topology without post-hoc feasibility fixes.

Ax Wenjing Chen, Chengyuan Qian, Shuo Xing, Yi Zhou, Victoria Crawford 3/10/2026

Multi-Agent Reinforcement Learning with Submodular Reward

Framework for cooperative multi-agent RL with submodular rewards modeling overlapping agent contributions. First formal analysis of diminishing returns in team coordination.

Ax Yanjun Chen, Yirong Sun, Hanlin Wang, Xinming Zhang, Xiaoyu Shen, Wenjie Li, Wei Zhang 3/10/2026

Contextual Counterfactual Credit Assignment for Multi-Agent Reinforcement Learning in LLM Collaboration

C3 algorithm for multi-agent reinforcement learning with LLMs. Addresses credit assignment problem in sparse reward scenarios through contextual counterfactual analysis.

Ax Mingi Kang, Zai Yang, Jeova Farias Sales Rocha Neto 3/10/2026

IGLU: The Integrated Gaussian Linear Unit Activation Function

Gaussian Linear Unit activation function analyzing mathematical relationships among modern transformer alternatives to ReLU.

Ax Abdulrahman Alswaidan, Jeffrey D. Varner 3/10/2026

Stochastic Attention via Langevin Dynamics on the Modern Hopfield Energy

Stochastic attention mechanism via Langevin sampling on Hopfield energy enabling temperature-controlled retrieval and generation.

Ax Gyujun Jeong (School of Electrical and Computer Engineering, Georgia Institute of Technology, GA, USA), Sungwon Cho (School of Electrical and Computer Engineering, Georgia Institute of Technology, GA, USA), Minji Shon (School of Electrical and Computer Engineering, Georgia Institute of Technology, GA, USA), Namhoon Kim (School of Electrical and Computer Engineering, Georgia Institute of Technology, GA, USA), Woohyun Hwang (Semiconductor Research and Development, Samsung Electronics Co., Ltd, South Korea), Kwangyou Seo (Semiconductor Research and Development, Samsung Electronics Co., Ltd, South Korea), Suhwan Lim (Semiconductor Research and Development, Samsung Electronics Co., Ltd, South Korea), Wanki Kim (Semiconductor Research and Development, Samsung Electronics Co., Ltd, South Korea), Daewon Ha (Semiconductor Research and Development, Samsung Electronics Co., Ltd, South Korea), Prasanna Venkatesan (NVIDIA, Santa Clara, CA, USA), Kihang Youn (NVIDIA, Santa Clara, CA, USA), Ram Cherukuri (NVIDIA, Santa Clara, CA, USA), Yiyi Wang (NVIDIA, Santa Clara, CA, USA), Suman Datta (School of Electrical and Computer Engineering, Georgia Institute of Technology, GA, USA), Asif Khan (School of Electrical and Computer Engineering, Georgia Institute of Technology, GA, USA), Shimeng Yu (School of Electrical and Computer Engineering, Georgia Institute of Technology, GA, USA) 3/10/2026

Physics-informed AI Accelerated Retention Analysis of Ferroelectric Vertical NAND: From Day-Scale TCAD to Second-Scale Surrogate Model

Physics-informed surrogate model for ferroelectric NAND retention analysis replacing expensive TCAD simulations.

Ax Yan-Ying Chen, Dule Shu, Matthew Hong, Andrew Taber, Jonathan Li, Matthew Klenk 3/10/2026

Learning From Design Procedure To Generate CAD Programs for Data Augmentation

LLM-based CAD program generation using design procedures and geometric constraints for parametric model synthesis.

Ax Jim Achterberg, Marcel Haas, Bram van Dijk, Marco Spruit 3/10/2026

XGenBoost: Synthesizing Small and Large Tabular Datasets with XGBoost

Generative models for tabular data using XGBoost as score estimator with denoising diffusion for small and large datasets.

Ax Nandan Kumar Jha, Brandon Reagen 3/10/2026

NerVE: Nonlinear Eigenspectrum Dynamics in LLM Feed-Forward Networks

Eigenspectral framework analyzing information flow in LLM feed-forward networks through lightweight spectral metrics.

Ax Zhixu Du, Krishna Teja Chitty-Venkata, Murali Emani, Venkatram Vishwanath, Hai Helen Li, Yiran Chen 3/10/2026

Swimba: Switch Mamba Model Scales State Space Models

Mixture-of-experts approach for state space models with expert specialization while maintaining computational efficiency.

Ax Milad Shirani, Pete H. Gueldner, Murat Khidoyatov, Jeremy L. Warren, Federica Ninno 3/10/2026

Physics-Consistent Neural Networks for Learning Deformation and Director Fields in Microstructured Media with Loss-Based Validation Criteria

Physics-consistent neural networks for Cosserat elasticity modeling deformation and director fields in microstructured materials.

Ax Ege C. Kaya, Mahsa Ghasemi, Abolfazl Hashemi 3/10/2026

Joint MDPs and Reinforcement Learning in Coupled-Dynamics Environments

Reinforcement learning formalism for coupled-dynamics environments specifying joint distributions across counterfactual actions.

Ax Yuhang Song, Naima Abrar Shami, Romaric Duvignau, Vasiliki Kalavri 3/10/2026

Not All Neighbors Matter: Understanding the Impact of Graph Sparsification on GNN Pipelines

Study of graph sparsification impact on GNN pipeline performance and scalability for billion-node graphs.

Ax Xin Zhang, Xingyu Li, Rongguang Wang, Ruizhong Miao, Zheng Wang, Dan Roth, Chenyang Li 3/10/2026

Chart-RL: Generalized Chart Comprehension via Reinforcement Learning with Verifiable Rewards

Reinforcement learning approach for chart comprehension in vision-language models using verifiable rewards for symbolic reasoning.

Ax Ruipeng Zhang, Hongzhan Yu, Ya-Chien Chang, Chenghao Li, Henrik I. Christensen, Sicun Gao 3/10/2026

Learning Quadruped Walking from Seconds of Demonstration

Imitation learning analysis for quadruped locomotion showing effectiveness in small data regimes via limit cycle structure.

Ax Jiwoo Yoon, Kyumin Choi, Jaewoong Choi 3/10/2026

Conditional Unbalanced Optimal Transport Maps: An Outlier-Robust Framework for Conditional Generative Modeling

Optimal transport framework for conditional generative modeling robust to outliers using unbalanced transport.

Ax Addison Kalanther, Sanika Bharvirkar, Shankar Sastry, Chinmay Maheshwari 3/10/2026

NePPO: Near-Potential Policy Optimization for General-Sum Multi-Agent Reinforcement Learning

Multi-agent reinforcement learning algorithm for general-sum games with heterogeneous agent objectives and convergence guarantees.

Ax Tong Yang, Moonkyung Ryu, Chih-Wei Hsu, Guy Tennenholtz, Yuejie Chi, Craig Boutilier, Bo Dai 3/10/2026

Diffusion Controller: Framework, Algorithms and Parameterization

Control-theoretic framework for controllable diffusion generation using linearly-solvable MDPs and reweighting pretrained models.

Ax Xiangjie Xiao, Cong Zhang, Wen Song, Zhiguang Cao 3/10/2026

RESCHED: Rethinking Flexible Job Shop Scheduling from a Transformer-based Architecture with Simplified States

Transformer-based approach to flexible job shop scheduling using simplified states instead of handcrafted features.

Ax Jiayi Wang, John Gounley, Heidi Hanson 3/10/2026

Resource-Adaptive Federated Text Generation with Differential Privacy

Federated learning method for generating differentially private synthetic text datasets from LLMs for downstream task reuse.

Ax Zhiji Yang, Mei Huang, Xinyu Li, Xianli Pan, Qi Wang, Jianhua Zhao 3/10/2026

Interpretable Maximum Margin Deep Anomaly Detection

Deep SVDD improvement for anomaly detection addressing hypersphere collapse and interpretability via maximum margin approach.

Ax Woogyeol Jin, Taywon Min, Yongjin Yang, Swanand Ravindra Kadhe, Yi Zhou, Dennis Wei, Nathalie Baracaldo, Kimin Lee 3/10/2026

Entropy-Aware On-Policy Distillation of Language Models

On-policy distillation method using entropy-aware objectives for improved knowledge transfer between language models.

Ax Michael Hauri, Friedemann Zenke 3/10/2026

Dreamer-CDP: Improving Reconstruction-free World Models Via Continuous Deterministic Representation Prediction

Dreamer-CDP improves world models with continuous deterministic representation prediction without reconstruction.

Ax Muhammad Khalifa, Zohaib Khan, Omer Tafveez, Hao Peng, Lu Wang 3/10/2026

Countdown-Code: A Testbed for Studying The Emergence and Generalization of Reward Hacking in RLVR

Benchmark environment for studying reward hacking in RL agents through dual-access mathematical reasoning tasks.

Ax Tao Shi, Liangming Chen, Long Jin, Mengchu Zhou 3/10/2026

Combining Adam and its Inverse Counterpart to Enhance Generalization of Deep Learning Optimizers

InvAdam optimizer variant that improves generalization by finding flatter minima than standard Adam.

Ax Subhojyoti Mukherjee, Stefano Petrangeli, Branislav Kveton, Trung Bui, Franck Dernoncourt, Arko Mukherjee 3/10/2026

Agentic Planning with Reasoning for Image Styling via Offline RL

AI agent framework using offline RL for structured planning and reasoning in image editing tasks.

Ax Yuxuan Han, Meng-Hao Guo, Zhengning Liu, Wenguang Chen, Shi-Min Hu 3/10/2026

Making LLMs Optimize Multi-Scenario CUDA Kernels Like Experts

LLM-based system for automated CUDA kernel optimization across ML and scientific computing domains.

Ax Haonan Xu, Yang Yang 3/10/2026

Shaping Parameter Contribution Patterns for Out-of-Distribution Detection

Method to improve OOD detection by diversifying parameter contribution patterns in classifiers.

Ax Jilong Liu, Yonghui Yang, Pengyang Shao, Haokai Ma, Wei Qin, Richang Hong 3/10/2026

wDPO: Winsorized Direct Preference Optimization for Robust LLM Alignment

wDPO improves DPO for LLM alignment by using winsorization to handle noisy preference data robustly.

Ax Chuxue Cao, Honglin Lin, Zhanping Zhong, Xin Gao, Mengzhang Cai, Conghui He, Sirui Han, Lijun Wu 3/10/2026

Unlocking Data Value in Finance: A Study on Distillation and Difficulty-Aware Training

Empirical study on knowledge distillation and difficulty-aware training for improving LLM performance in finance domain.

Ax Andrea Giuseppe Di Francesco, Andrea Rubbi, Pietro Li\`o 3/10/2026

Retrieval-Augmented Generation for Predicting Cellular Responses to Gene Perturbation

PT-RAG uses retrieval-augmented generation to predict cellular responses to gene perturbations with improved generalization.

Ax Zixuan Yu, Zhenheng Tang, Tongliang Liu, Chengqi Zhang, Xiaowen Chu, Bo Han 3/10/2026

Rethinking Deep Research from the Perspective of Web Content Distribution Matching

WeDas framework improves web search agents by matching queries to web content distribution structures for better evidence retrieval.

Ax Madhurima Panja, Grace Younes, Tanujit Chakraborty 3/10/2026

Turning Time Series into Algebraic Equations: Symbolic Machine Learning for Interpretable Modeling of Chaotic Time Series

Symbolic machine learning method to convert chaotic time series into interpretable algebraic equations for forecasting.

Ax Nilesh Jain, Rohit Yadav, Sagar Kotian, Claude AI 3/10/2026

AutoResearch-RL: Perpetual Self-Evaluating Reinforcement Learning Agents for Autonomous Neural Architecture Discovery

AutoResearch-RL is an RL agent that autonomously conducts perpetual neural architecture and hyperparameter search via code modification without human supervision.

Ax Angad Singh Ahuja 3/10/2026

Adversarial Latent-State Training for Robust Policies in Partially Observable Domains

Adversarial latent-state training framework for robust policies in partially observable MDPs under latent distribution shift with theoretical guarantees.

Ax Lujing Zhang, Daniel Hsu, Sivaraman Balakrishnan 3/10/2026

ShakyPrepend: A Multi-Group Learner with Improved Sample Complexity

ShakyPrepend applies differential privacy-inspired tools to multi-group learning for improved sample complexity and adaptation to group structure.

Ax Truong Xuan Khanh, Truong Quynh Hoa 3/10/2026

Norm-Hierarchy Transitions in Representation Learning: When and Why Neural Networks Abandon Shortcuts

Analyzes norm-hierarchy transitions explaining when neural networks transition from spurious shortcuts to structured representations during training.

Ax Antonio De Santis, Schrasing Tong, Marco Brambilla, Lalana Kagal 3/10/2026

Learning Concept Bottleneck Models from Mechanistic Explanations

Learning concept bottleneck models from mechanistic explanations instead of pre-specified or LLM-prompted concepts for improved interpretability and predictive power.

Ax Yuanyun Zhang, Shi Li 3/10/2026

Learning Clinical Representations Under Systematic Distribution Shift

Addresses representation entanglement between physiologic signal and institutional artifacts in clinical ML under systematic distribution shift from heterogeneous practices.

Ax Sean Gunn, Jorio Cocola, Oliver De Candido, Vaggos Chatziafratis, Paul Hand 3/10/2026

Latent Generative Models with Tunable Complexity for Compressed Sensing and other Inverse Problems

Develops tunable-complexity priors for diffusion models and normalizing flows to balance representation error and overfitting in inverse problem solving.

Ax Yucheng Xing, Xin Wang 3/10/2026

N-Tree Diffusion for Long-Horizon Wildfire Risk Forecasting

N-Tree Diffusion enables efficient long-horizon wildfire risk forecasting by hierarchically extending diffusion models across multiple prediction steps.

Ax Mohammed Alnemari, Rizwan Qureshi, Nader Begrazadah 3/10/2026

Scaling Laws in the Tiny Regime: How Small Models Change Their Mistakes

Examines neural scaling laws in sub-20M parameter regime for TinyML/edge AI, showing both ConvNets and MobileNetV2 follow power law error scaling.

Ax Hieu Le, Oguz Bedir, Mostafa Ibrahim, Jian Tao, Sabit Ekin 3/10/2026

Learning to Reflect: Hierarchical Multi-Agent Reinforcement Learning for CSI-Free mmWave Beam-Focusing

Hierarchical multi-agent RL framework for controlling reconfigurable intelligent surfaces in mmWave systems without channel state information estimation overhead.

Ax Xuxing Chen, Yun He, Jiayi Xu, Minhui Huang, Xiaoyi Liu, Boyang Liu, Fei Tian, Xiaohan Wei, Rong Jin, Sem Park, Bo Long, Xue Feng 3/10/2026

Feed m Birds with One Scone: Accelerating Multi-task Gradient Balancing via Bi-level Optimization

Accelerates multi-task learning gradient balancing through bi-level optimization to improve MGDA-type methods for handling task conflicts.

Ax Rian Atri 3/10/2026

Deterministic Fuzzy Triage for Legal Compliance Classification and Evidence Retrieval

Deterministic fuzzy triage system for legal compliance classification using dual encoders and transparent bands, demonstrated on contractual evidence HIPAA/NERC-CIP alignment.