Stratification and Survival Prediction for Amyotrophic Lateral Sclerosis Patients. - Yixiao Huang 004; Xiaoli Wu; Rosa H. M. Chan (BHI, 2022)

Mechanics of Next Token Prediction with Self-Attention. - Yingcong Li; Yixiao Huang 004; Muhammed Emrullah Ildiz; Ankit Singh Rawat; Samet Oymak (AISTATS, 2024)

From Self-Attention to Markov Models: Unveiling the Dynamics of Generative Transformers. - Muhammed Emrullah Ildiz; Yixiao Huang 004; Yingcong Li; Ankit Singh Rawat; Samet Oymak (ICML, 2024)

Towards Efficient Training and Evaluation of Robust Models against l0 Bounded Adversarial Perturbations. - Xuyang Zhong; Yixiao Huang 004; Chen Liu 027 (ICML, 2024)

Towards Efficient Training and Evaluation of Robust Models against l0 Bounded Adversarial Perturbations. - Xuyang Zhong; Yixiao Huang 004; Chen Liu 027 (CoRR, 2024)

On the Power of Convolution-Augmented Transformer. - Mingchen Li; Xuechen Zhang 007; Yixiao Huang 004; Samet Oymak (AAAI, 2025)

Fast Adversarial Training against Sparse Attacks Requires Loss Smoothing. - Xuyang Zhong; Yixiao Huang 004; Chen Liu 027 (CoRR, 2025)

OVERT: A Benchmark for Over-Refusal Evaluation on Text-to-Image Models. - Ziheng Cheng; Yixiao Huang 004; Hui Xu; Somayeh Sojoudi; Xuandong Zhao; Dawn Song; Song Mei (CoRR, 2025)

Generalization or Hallucination? Understanding Out-of-Context Reasoning in Transformers. - Yixiao Huang 004; Hanlin Zhu; Tianyu Guo 004; Jiantao Jiao; Somayeh Sojoudi; Michael I. Jordan; Stuart Russell 001; Song Mei (CoRR, 2025)

Spooky Action at a Distance: Normalization Layers Enable Side-Channel Spatial Communication. - Samuel Pfrommer; George Ma; Yixiao Huang 004; Somayeh Sojoudi (CoRR, 2025)

Reinforcement Learning for Flow-Matching Policies. - Samuel Pfrommer; Yixiao Huang 004; Somayeh Sojoudi (CoRR, 2025)