Publications and Reports

Yatong Bai, University of California, Berkeley

Preprint Papers

  1. MixedNUTS: Training-Free Accuracy-Robustness Balance via Nonlinearly Mixed Classifiers,
    ArXiv preprint arXiv:2402.02263, 2024.
    Yatong Bai, Mo Zhou, Vishal M. Patel, and Somayeh Sojoudi.
    Project code available at github.com/Bai-YT/MixedNUTS.

  2. ConsistencyTTA: Accelerating Diffusion-Based Text-to-Audio Generation with Consistency Distillation,
    ArXiv preprint arXiv:2309.10740, 2023.
    Yatong Bai, Trung Dang, Dung Tran, Kazuhito Koishida, and Somayeh Sojoudi.
    Project website consistency-tta.github.io.
    Project code available at github.com/Bai-YT/ConsistencyTTA.
    Model checkpoints available at huggingface.co/Bai-YT/ConsistencyTTA.
    Result presented on PapersWithCode.
    Work done during internship at Microsoft (Applied Science Group).

  3. Let's Go Shopping (LGS) — Web-Scale Image-Text Dataset for Visual Concept Understanding,
    ArXiv preprint arXiv:2401.04575, 2024.
    Yatong Bai, Utsav Garg, Erhan Bas, Isidora Filipovic, Apaar Shanker, Haoming Zhang, Samyak Parajuli, Amelia N. Chu, Eugenia D Fomitcheva, Elliot Branson, Aerin Kim, Somayeh Sojoudi, and Kyunghyun Cho.
    Work done during internship at Scale AI.

Journal Papers

  1. Improving the Accuracy-Robustness Trade-Off of Classifiers via Adaptive Smoothing,
    ArXiv preprint arXiv:2301.12554, 2023.
    Yatong Bai, Brendon G. Anderson, Aerin Kim, and Somayeh Sojoudi.
    Project code available at github.com/Bai-YT/AdaptiveSmoothing.
    SOTA results and model checkpoints verified and listed on RobustBench.

  2. Efficient Global Optimization of Two-Layer ReLU Networks: Adversarial Training and Quadratic-time Algorithms,
    in SIAM Journal on Mathematics of Data Science (SIMODS), 2022.
    Yatong Bai, Tanmay Gautam, and Somayeh Sojoudi.
    2021 INFORMS Data Mining Best Student Paper Award Runner-Up.

Conference Papers

  1. Initial State Interventions for Deconfounded Imitation Learning,
    in the IEEE Conference on Decision and Control (CDC), 2023.
    Samuel Pfrommer, Yatong Bai, Hyunin Lee, Somayeh Sojoudi.

  2. Mixing Classifiers to Alleviate the Accuracy-Robustness Trade-Off, 2023.
    Yatong Bai, Brendon G. Anderson, and Somayeh Sojoudi.

  3. Practical Convex Formulation of Robust One-Hidden-Layer Neural Network Training,
    in American Control Conference (ACC), 2022.
    Yatong Bai, Tanmay Gautam, Yu Gai, and Somayeh Sojoudi.

Technical Reports

  1. Imitation Human Behavior Model for Interactive Autonomous Driving Simulation.
    Chenran Li, Akio Kodaira, Yatong Bai.
    Course project for Fall 2021 CS 285 — Deep Reinforcement Learning (UC Berkeley).

  2. Instrument Identification for Real-world Music Pieces.
    Yatong Bai, Zhubo Zhou, Tianyi Zuo, Zhaoran Ma, Junzhe Ruan.
    Course project for Summer 2020 CS 4641 — Machine Learning (Georgia Tech).
    Project code available at github.com/Bai-YT/CS4641-Music-Instrument-Recognition.

  3. Design and Implementation of a Data-Display MATLAB GUI for Cochlear Model Simulator.
    Yatong Bai, 2018.