Ax Zhiying Jiang, Raihan Seraj, Marcos Villagra, Bidhan Roy 3/10/2026

Heterogeneous Decentralized Diffusion Models

Efficient decentralized framework for training diffusion models with heterogeneous objectives across isolated experts with reduced computational requirements.

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 Serio Agriesti (Department of Technology, Management and Economics, Technical University of Denmark, Lyngby, Denmark), Guido Cantelmo (Department of Technology, Management and Economics, Technical University of Denmark, Lyngby, Denmark), Francisco Camara Pereira (Department of Technology, Management and Economics, Technical University of Denmark, Lyngby, Denmark) 3/10/2026

Metalearning traffic assignment for network disruptions with graph convolutional neural networks

Meta-learning approach using graph convolutional networks to handle traffic flow prediction under network disruptions.

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 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 Dhruman Gupta (Ashoka University), Yashas Shende (Ashoka University), Aritra Das (Ashoka University), Chanda Grover Kamra (Ashoka University), Debayan Gupta (Ashoka University) 3/10/2026

Joint 3D Gravity and Magnetic Inversion via Rectified Flow and Ginzburg-Landau Guidance

Rectified flow and Ginzburg-Landau guidance methods for 3D gravity and magnetic inversion in subsurface ore detection. Physics-based inverse problem solving.

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