On conditional Sibson's $ alpha$ -Mutual Information. - Amedeo Roberto Esposito; Diyuan Wu; Michael Gastpar (ISIT, 2021)

On conditional Sibson's α-Mutual Information. - Amedeo Roberto Esposito; Diyuan Wu; Michael Gastpar (CoRR, 2021)

Mean-field analysis for heavy ball methods: Dropout-stability, connectivity, and global convergence. - Diyuan Wu; Vyacheslav Kungurtsev; Marco Mondelli (Trans. Mach. Learn. Res., 2023)

The Iterative Optimal Brain Surgeon: Faster Sparse Recovery by Leveraging Second-Order Information. - Diyuan Wu; Ionut-Vlad Modoranu; Mher Safaryan; Denis Kuznedelev; Dan Alistarh (NeurIPS, 2024)

Neural Collapse Beyond the Unconstrainted Features Model: Landscape, Dynamics, and Generalization in the Mean-Field Regime. - Diyuan Wu; Marco Mondelli (CoRR, 2025)

Attention with Trained Embeddings Provably Selects Important Tokens. - Diyuan Wu; Aleksandr Shevchenko; Samet Oymak; Marco Mondelli (CoRR, 2025)