Ax Atharva Hans, Ilias Bilionis 16d ago

Coding-agents can replicate scientific machine learning papers

Coding agents replicate scientific ML paper claims autonomously. Framework validates agent ability to verify computational results from paper materials.

Ax Joe Watson, Joana Ribeiro de Faria, Marcus Tomalin, M{\aa}ns Magnusson, Huiyuan Xie, Hao Tian Yeung, Christine Carter, Jonathan Rutherford, Felix Steffek 16d ago

Shortcut Learning in Legal Judgment Prediction: Empirical Evidence from the UK Employment Tribunal

Investigates shortcut learning in legal judgment prediction using UK Employment Tribunal data. Domain-specific ML study with limited tech relevance.

Ax Wachiravit Modecrua, Krittin Pachtrachai, Touchapon Kraisingkorn 16d ago

Large Behavior Model: A Promptable Digital Twin of the Retail Customer

Large Behavioral Model learns customer decision-making from retail transaction data. Domain-specific application with limited broader ML significance.

Ax Emanuele Zappala 16d ago

Projection Methods for Operator Learning and Universal Approximation

Universal approximation theorem for operators on Banach spaces using projection methods. Theoretical ML research with limited practical applications.

Ax Duy Nguyen, Archiki Prasad, Elias Stengel-Eskin, Mohit Bansal 16d ago

Multi-Attribute Steering of Language Models via Targeted Intervention

Inference-time intervention method for steering LLM behavior across multiple conflicting attributes without parameter updates. Addresses multi-attribute alignment challenges.

Ax Jiancheng Huang, Gengwei Zhang, Zequn Jie, Siyu Jiao, Yinlong Qian, Ling Chen, Yunchao Wei, Lin Ma 16d ago

M4V: Multimodal Mamba for Efficient Text-to-Video Generation

M4V uses Mamba architecture for efficient text-to-video generation with linear-time sequence modeling.

Ax Duy Nguyen, Archiki Prasad, Elias Stengel-Eskin, Mohit Bansal 16d ago

GrAInS: Gradient-based Attribution for Inference-Time Steering of LLMs and VLMs

GrAInS enables gradient-based inference-time steering of LLMs and VLMs without weight updates.

Ax Skatje Myers, Dmitriy Dligach, Timothy A. Miller, Samantha Barr, James Landefeld, Yanjun Gao, Matthew Churpek, Anoop Mayampurath, Majid Afshar 16d ago

Evaluating Retrieval-Augmented Generation vs. Long-Context Input for Clinical Reasoning over EHRs

Evaluation of RAG versus long-context prompting for clinical reasoning tasks over electronic health records.

Ax Mengjie Li, Yuan Feng, Xike Xie, William J. Song 16d ago

REAL: REtrieval-reAsoning and Logic-constructed Attention Behaviors for Long-Context KV Cache Compression

REAL method for KV cache compression in long-context LLMs via retrieval-reasoning and attention analysis.

Ax Houcheng Jiang, Junfeng Fang, Jiaxin Wu, Tianyu Zhang, Chen Gao, Xiang Wang, Xiangnan He, Yang Deng 16d ago

Contrastive Weak-to-strong Generalization

Weak-to-strong generalization for LLMs using contrastive learning with implicit rewards to improve robustness.

Ax Georgios Anyfantis, Pere Barlet-Ros 16d ago

AutoGraphAD: Unsupervised network anomaly detection using Variational Graph Autoencoders

Unsupervised network anomaly detection using variational graph autoencoders without requiring labeled datasets.

Ax Scott Merrill, Shashank Srivastava 16d ago

Point of Order: Action-Aware LLM Persona Modeling for Data-Grounded Civic Deliberation

Pipeline for speaker-attributed LLM persona modeling from civic deliberation recordings for controlled simulations.

Ax Natalia Ponomareva, Zheng Xu, H. Brendan McMahan, Peter Kairouz, Lucas Rosenblatt, Vincent Cohen-Addad, Crist\'obal Guzm\'an, Ryan McKenna, Galen Andrew, Alex Bie, Da Yu, Alex Kurakin, Morteza Zadimoghaddam, Sergei Vassilvitskii, Andreas Terzis 16d ago

How to DP-fy Your Data: A Practical Guide to Generating Synthetic Data With Differential Privacy

Practical guide for generating synthetic data with differential privacy to address data scarcity and representation issues.

Ax Zihan Zhou, Animesh Garg, Ajay Mandlekar, Caelan Garrett 16d ago

ReinforceGen: Hybrid Skill Policies with Automated Data Generation and Reinforcement Learning

ReinforceGen system combines task decomposition, imitation learning, and RL fine-tuning for long-horizon robotic manipulation.

Ax Tanmay Ambadkar, Sourav Panda, Shreyash Kale, Jonathan Dodge, Abhinav Verma 16d ago

Preference Conditioned Multi-Objective Reinforcement Learning: Decomposed, Diversity-Driven Policy Optimization

MORL approach using preference-conditioned policies to recover dense Pareto fronts by addressing early scalarization and advantage cancellation issues.

Ax Zirong Chen, Meiyi Ma 16d ago

Empowering 9-1-1 Calltaking Training with Generative AI: Experiences and Lessons Learned

arXiv paper on deploying generative AI for 911 call-taker training to address staffing and training scalability challenges.

Ax Zac Garby, Andrew D. Gordon, David Sands 16d ago

The LLMbda Calculus: AI Agents, Conversations, and Information Flow

arXiv paper on LLMbda Calculus: formal framework for information flow control in LLM agents to defend against prompt injection.

Ax Gangda Deng, Zhaoling Chen, Zhongming Yu, Haoyang Fan, Yuhong Liu, Yuxin Yang, Dhruv Parikh, Rajgopal Kannan, Le Cong, Mengdi Wang, Qian Zhang, Viktor Prasanna, Xiangru Tang, Xingyao Wang 16d ago

SWE-Milestone: Evaluating AI Agents on Continuous Software Evolution

arXiv paper on SWE-Milestone: benchmark for evaluating AI agents on continuous software evolution with temporal dependencies.

Ax Iakovos-Christos Zarkadis, Christos Douligeris 16d ago

Machine Learning for Network Attacks Classification and Statistical Evaluation of Adversarial Learning Methodologies for Synthetic Data Generation

arXiv paper on ML for network attack classification and synthetic data generation using adversarial methods.

Ax Harshit Joshi, Priyank Shethia, Jadelynn Dao, Monica S. Lam 16d ago

SLIDERS: Systematic Reviews via Automated Evidence Synthesis and Reconciliation

arXiv paper on SLIDERS: LLM-based systematic evidence synthesis and reconciliation for comprehensive document analysis.

Ax Filip Edstr\"om, Guilherme W. F. Barros, Tetiana Gorbach, Xavier de Luna 16d ago

Tuning Derivatives for Causal Fairness in Machine Learning

arXiv paper on causal fairness in ML by tuning derivatives to handle bias in protected attributes.

Ax Vardhan Dongre, Dilek Hakkani-T\"ur 16d ago

Embodied Multi-Agent Coordination by Aligning World Models Through Dialogue

arXiv paper on embodied multi-agent coordination using dialogue to align partially-observable world models.

Ax Tao Xie, Zexi Tan, Haoyi Xiao, Mengke Li, Yiqun Zhang, Yang Lu, Cuie Yang, Yiu-ming Cheung 16d ago

AnchorMoE: Interpretable Time Series Classification via Anchor-Routed MoE

arXiv paper on AnchorMoE: interpretable multivariate time series classification using mixture-of-experts routing.

Ax Ali Behrouz, Farnoosh Hashemi, Adel Javanmard, Vahab Mirrokni 16d ago

Language Models Need Sleep: Learning to Self-Modify and Consolidate Memories

arXiv paper on enabling LLMs to self-modify and consolidate memories for continual learning beyond in-context knowledge.

Ax Tianming Du, Peijie Yu, Sihan Shang, Danli Shi, My Linh Nguyen, Shengbo Gao, Guangyuan Li, Yinghong Yu, Yan Jiang, Qianlong Zhao, Behzad Bozorgtabar, Shaoxiong Ji, Jiazhen Pan, Daniel Rueckert, Jiancheng Yang 16d ago

Are LLMs Ready to Assist Physicians? PhysAssistBench for Interactive Doctor-Patient-EHR Assistance

arXiv paper on PhysAssistBench: benchmark for evaluating LLMs assisting physicians through coordinated clinical knowledge, EHR interaction, and patient communication.

Ax Zijun Xie, Binbin Zheng, Enlei Gong, Jihua Liu, Yuyang You, Lingfeng Liu, Jiayao Tang, Guanqun Zhao, Aoqi Hu, Zeyu Chen 16d ago

ECHO: Prune To Act, Trace To Learn With Selective Turn Memory In Agentic RL

arXiv paper on ECHO: selective turn memory and pruning techniques for long-horizon language agents with bounded context windows.

Ax Diego Cerda-Mardini, Sarath Chandar, Sreenath Madathil 16d ago

Consistent but Miscalibrated: Evaluating LLM Limitations for Risk Communication in Natural Language

arXiv paper evaluating 9 LLMs' ability to communicate probabilistic information in natural language with consistency and calibration assessment.

Ax Abhishek Kumar, Carsten Maple 16d ago

Refused in Chat, Written in Code: Workflow-Level Jailbreak Construction in IDE Coding Agents

arXiv paper on workflow-level jailbreak construction in IDE coding agents, showing safety failures across multi-turn task decomposition.

Ax Haz Sameen Shahgir, Yufei Li, Xiaohan Wei, Yunchen Pu, Fei Tian, Chonglin Sun, Frank Shyu, Sandeep Pandey, Luke Simon, Yue Dong, Xi Liu 16d ago

SCOReD: Student-Aware CoT Optimization for Recommendation Distillation

arXiv paper on chain-of-thought distillation optimization for recommendation systems using student-aware techniques.

Ax Andrea Bacciu, Andrea Alfarano, Saab Mansour, Amin Mantrach, Marcello Federico 16d ago

Estimating Uncertainty from Reasoning: A Large-Scale Study of Multi- and Crosslingual MCQA Performance in LLMs

Large-scale evaluation of uncertainty estimation methods across 22 languages in LLMs for multi-choice question answering.

Ax Yusen Feng, Bingchen Han, Jiangran Lyu, Kai Liu, Yixin Zheng, Yuxuan Wan, Weiheng Liu, Sun Han, Ruiqin Li, Yulong Zhang, Fangfu Liu, Xuesong Shi, Libin Liu, Yizhou Wang, Zhizheng Zhang, He Wang 16d ago

WAM-TTT: Steering World-Action Models by Watching Human Play at Test Time

Test-time training framework for steering robot foundation models toward task variants using human video demonstrations without fine-tuning.

Ax Szczepan Konior, Alexandre Quemy, Przemys{\l}aw Klocek, Bart{\l}omiej Sobieski, Gr\'egoire Cattan 16d ago

Riemannian Geometry for Pre-trained Language Model Embeddings

Method extracting Riemannian geometric structure from pre-trained language model embeddings to understand sentence classification geometry.

Ax Zhoujie Hou, Song Wang, Kexin Lou, Mo Wang, Chen Wei, Quanying Liu 16d ago

Omni-Sleep: A Sleep Foundation Model via Hierarchical Contrastive Learning of CNS-ANS Dynamics

Foundation model for sleep analysis using hierarchical contrastive learning on multimodal biosignals from CNS and ANS.

Ax Haozhan Tang, Zerui Wang, Yuxian Gu, Song Han, Han Cai 16d ago

Jet-Long: Efficient Long-Context Extension with Dynamic Bifocal RoPE

Efficient zero-shot context extension method for LLMs using dynamic bifocal RoPE to handle long-context applications without retraining.

Ax Qingzhuo Wang, Ruiyang Qin, Zhenxin Qin, Wen Shen, Zhihua Wei 16d ago

A Unified Approach to Interpreting Knowledge Distillation for Large Language Models via Interactions

Framework interpreting knowledge distillation mechanisms in LLMs via interaction decomposition to understand why various KD methods succeed.

Ax Farica Zhuang, Seong Woo Han, Zixuan Wen, Shu Yang, Yize Zhao, Li Shen 16d ago

iLENS: Interpretable LLM-Guided Mixture-of-Experts for Neuroimaging Survival Analysis

System combining LLM-guided Mixture-of-Experts with survival analysis for interpretable Alzheimer's disease risk prediction from neuroimaging.

Ax Ian Colbert, Eashan Dash, Pablo Monteagudo-Lago, Juan Amboage, Srinidhi N, Giuseppe Franco, Nicholas J. Fraser, Arun Ramachandran 16d ago

Signed Symmetric Quantization for Few-Bit Integers

Quantization technique adjusting scale asymmetry for few-bit integer precision to reduce clipping errors on outliers.

Ax Ali Kayyam 16d ago

Sticky Routing: Training MoE Models for Memory-Efficient Inference

Training method for Mixture-of-Experts models reducing memory-access overhead on edge devices via differentiable routing consistency loss.

Ax Kehan Guo, Yili Shen, Yujun Zhou, Yue Huang, Chujie Gao, Shiyi Du, Xiangliang Zhang 16d ago

Reward Transport: Property Control in Flow Matching via Noise-Space Alignment

Flow matching method using optimal transport coupling for controlled generation of molecules with target properties.

Ax Qianli Liu, Kaibin Guo, Zicong Hong, Peng Li, Fahao Chen, Haodong Wang, Jian Lin, Song Guo 16d ago

Director: Accelerating Distributed MoE Serving via Online Proactive Expert Placement

System for optimizing expert placement in distributed Mixture-of-Experts model serving via online proactive placement strategy.

Ax Thinh T. H. Nguyen, Le-Tuan Nguyen, Minh-Duong Nguyen, Nhi Trinh, Anh Tran Nam Nguyet, Dung D. Le, Kok-Seng Wong 16d ago

HERO: A Heterogeneity-Aware Benchmark Library for Federated Continual Learning

Benchmark library for federated continual learning with standardized evaluation protocol across datasets, task splits, and data distributions.

Ax Qiheng Sun, Hongwei Zhang, Junxu Liu, Xiaokai Mao, Jinfei Liu, Kui Ren, Haibo Hu 16d ago

DaDaDa: A Dataset for Data Pricing in Data Marketplaces

Dataset for studying data pricing mechanisms in data marketplaces, addressing valuation challenges for data products.

Ax Tao Lu, Haoyu Wang, Zonghui Wang, Keshen Xiang, Jiaheng Zhang, Wenzhi Chen 16d ago

Accelerating GPU Inference of Large Language Models with Moderately Unstructured Sparse Weight Matrices

GPU kernel optimization for LLM inference acceleration using moderately sparse weight matrices, addressing performance gaps in sparse matrix multiplication.

Ax Sunshine Jiang, John Marangola, David Zhang, Raghuram Kowdeed, Ruiyang Luo, Nitish Dashora, Richard Li, Pulkit Agrawal, Zhang-Wei Hong 16d ago

Prompt-Driven Exploration

Method using LLMs and vision-language-action models to improve exploration in reinforcement learning by generating diverse policy perturbations via natural language prompts.

Ax William W. Yang, Andrew M. Saxe, Peter E. Latham 16d ago

How are linear representations learned? Exact solutions to the dynamics of abstraction

Research on how linear representations emerge during neural network training, foundational for interpretability methods like linear probes and activation steering.

Ax Ege \c{C}akar, Hannah Guan, Kayden Kehe 16d ago

Optimizing Against Safety Representations: Activation-Guided Adversarial Suffixes and the Geometry of Refusal

Study of adversarial attacks on LLM safety representations via activation-guided optimization and refusal geometry.

Ax Minett Tran, Taehee Jeong 16d ago

Pattern-Aware Graph Neural Networks for Handling Missing Data

GNN approach that encodes missing data patterns alongside values for handling incomplete datasets.

Ax Yuanjie Zhu, Liangwei Yang, Ke Xu, Weizhi Zhang, Shanghao Li, Zihe Song, Philip S. Yu 16d ago

BlockServe: Block-Grained Continuous Batching for High-Throughput Diffusion LLM Serving

Continuous batching framework for efficient diffusion LLM serving with block-grained scheduling to reduce latency.

Ax Fangxu Yu, Tao Feng, Dehai Min, Lu Cheng, Ge Liu, Tianyi Zhou 16d ago

TSRouter: Dynamic Modality-Model Selection for Time Series Reasoning

Dynamic routing system selecting between LLMs and VLMs for time series reasoning based on modality characteristics.

Ax Nick Souligne, Isabella Mixton-Garcia, Vignesh Subbian 16d ago

FairSelect: A Systematic Evaluation of Multi-Level and Intersectional Algorithmic Fairness

Toolkit for systematic evaluation of algorithmic fairness across intersectional subgroups and modeling lifecycle stages.