Ax Theodora Worledge, Othman Bensouda Koraichi, Daniel Bernal, Aviv Caspi, Tatsunori Hashimoto, Carlos Guestrin, David Freeman Engstrom 9d ago

AI Assistance for Human Review of Default Judgments

Study of AI assistance for reviewing default judgments in US courts, finding high error rates in debt collection cases.

Ax Paimon Goulart, Chansong Lim, N\'icolas Roque dos Santos, Yue Dong, Sheldon Peterson, Jia Chen, Evangelos E. Papalexakis 9d ago

Multi-modal Rail Crossing Safety Analysis

Multi-modal system combining images and structured accident data to assess railway crossing safety.

Ax Kathan Shah 9d ago

Token Geometry

Research on gradient geometry of embedding tables in language models; introduces Ember optimizer for efficient finetuning and pretraining with minimal memory.

Ax Jiatong Li, Samuel Yeh, Sharon Li 9d ago

Multi-Head Recurrent Memory Agents

Multi-Head Recurrent Memory Agents: Addresses reliability degradation in LLM agents with long contexts by decomposing memory capture and retention.

Ax Ege Onur Taga, Yilin Zhuang, M. Emrullah Ildiz, Petros Mol, Abhimanyu Das, Karthik Duraisamy, Samet Oymak 9d ago

Evolutionary Feature Engineering for Structured Data

EFE: Framework using LLM-based evolutionary search to discover preprocessing transformations for structured data as composable Python programs.

Ax Zhaoyan Sun, Shan Zhong, Daizhou Wen, Jiaxing Han, Guoliang Li, Ying Yan, Peng Zhang, Yu Su, Xiang Qi, Baolin Sun, Chengyuan Yang, Tao Fang, Huaiyu Ruan 9d ago

AgenticDataBench: A Comprehensive Benchmark for Data Agents

Comprehensive benchmark for LLM-based data agents automating data science workflows. Evaluates agent capabilities on heterogeneous datasets.

Ax Stefano Masini, Cecilia Viscardi, Michela Baccini 9d ago

Full Bayesian Reinforcement Learning via LF-IBIS

Bayesian reinforcement learning method addressing data scarcity through prior knowledge and belief updates in sequential decision-making.