Practical Convex Formulation of Robust One-hidden-layer Neural Network Training. - Yatong Bai; Tanmay Gautam; Yu Gai; Somayeh Sojoudi (CoRR, 2021)

Practical Convex Formulations of One-hidden-layer Neural Network Adversarial Training. - Yatong Bai; Tanmay Gautam; Yu Gai; Somayeh Sojoudi (ACC, 2022)

Initial State Interventions for Deconfounded Imitation Learning. - Samuel Pfrommer; Yatong Bai; Hyunin Lee; Somayeh Sojoudi (CDC, 2023)

Accelerating Diffusion-Based Text-to-Audio Generation with Consistency Distillation. - Yatong Bai; Trung Dang 002; Dung N. Tran; Kazuhito Koishida; Somayeh Sojoudi (CoRR, 2023)

Efficient Global Optimization of Two-Layer ReLU Networks: Quadratic-Time Algorithms and Adversarial Training. - Yatong Bai; Tanmay Gautam; Somayeh Sojoudi (SIAM J. Math. Data Sci., 2023)

Ranking Manipulation for Conversational Search Engines. - Samuel Pfrommer; Yatong Bai; Tanmay Gautam; Somayeh Sojoudi (EMNLP, 2024)

ConsistencyTTA: Accelerating Diffusion-Based Text-to-Audio Generation with Consistency Distillation. - Yatong Bai; Trung Dang 002; Dung N. Tran; Kazuhito Koishida; Somayeh Sojoudi (INTERSPEECH, 2024)

Mixing classifiers to alleviate the accuracy-robustness trade-off. - Yatong Bai; Brendon G. Anderson; Somayeh Sojoudi (L4DC, 2024)

Let's Go Shopping (LGS) - Web-Scale Image-Text Dataset for Visual Concept Understanding. - Yatong Bai; Utsav Garg; Apaar Shanker; Haoming Zhang; Samyak Parajuli; Erhan Bas; Isidora Filipovic; Amelia N. Chu; Eugenia D. Fomitcheva; Elliot Branson; Aerin Kim; Somayeh Sojoudi; et al. (CoRR, 2024)

Improving the Accuracy-Robustness Trade-Off of Classifiers via Adaptive Smoothing. - Yatong Bai; Brendon G. Anderson; Aerin Kim; Somayeh Sojoudi (SIAM J. Math. Data Sci., 2024)

MixedNUTS: Training-Free Accuracy-Robustness Balance via Nonlinearly Mixed Classifiers. - Yatong Bai; Mo Zhou; Vishal M. Patel; Somayeh Sojoudi (Trans. Mach. Learn. Res., 2024)

DRAGON: Distributional Rewards Optimize Diffusion Generative Models. - Yatong Bai; Jonah Casebeer; Somayeh Sojoudi; Nicholas J. Bryan (CoRR, 2025)