| 2025 | ICLR | Flow matching achieves almost minimax optimal convergence. | Kenji Fukumizu, Taiji Suzuki, Noboru Isobe, Kazusato Oko, Masanori Koyama |
| 2025 | ICLR | Direct Distributional Optimization for Provable Alignment of Diffusion Models. | Ryotaro Kawata, Kazusato Oko, Atsushi Nitanda, Taiji Suzuki |
| 2025 | ICML | Nonlinear transformers can perform inference-time feature learning. | Naoki Nishikawa, Yujin Song, Kazusato Oko, Denny Wu, Taiji Suzuki |
| 2024 | COLT | Learning sum of diverse features: computational hardness and efficient gradient-based training for ridge combinations. | Kazusato Oko, Yujin Song, Taiji Suzuki, Denny Wu |
| 2024 | ICLR | Symmetric Mean-field Langevin Dynamics for Distributional Minimax Problems. | Juno Kim, Kakei Yamamoto, Kazusato Oko, Zhuoran Yang, Taiji Suzuki |
| 2024 | ICLR | Improved statistical and computational complexity of the mean-field Langevin dynamics under structured data. | Atsushi Nitanda, Kazusato Oko, Taiji Suzuki, Denny Wu |
| 2024 | ICML | SILVER: Single-loop variance reduction and application to federated learning. | Kazusato Oko, Shunta Akiyama, Denny Wu, Tomoya Murata, Taiji Suzuki |
| 2024 | ICML | Mean Field Langevin Actor-Critic: Faster Convergence and Global Optimality beyond Lazy Learning. | Kakei Yamamoto, Kazusato Oko, Zhuoran Yang, Taiji Suzuki |
| 2023 | ICALP | Nearly Tight Spectral Sparsification of Directed Hypergraphs. | Kazusato Oko, Shinsaku Sakaue, Shin-ichi Tanigawa |
| 2023 | ICML | Primal and Dual Analysis of Entropic Fictitious Play for Finite-sum Problems. | Atsushi Nitanda, Kazusato Oko, Denny Wu, Nobuhito Takenouchi, Taiji Suzuki |
| 2023 | ICML | Diffusion Models are Minimax Optimal Distribution Estimators. | Kazusato Oko, Shunta Akiyama, Taiji Suzuki |
| 2022 | ICLR | Particle Stochastic Dual Coordinate Ascent: Exponential convergent algorithm for mean field neural network optimization. | Kazusato Oko, Taiji Suzuki, Atsushi Nitanda, Denny Wu |
| 2022 | MOBIHOC | MOCHA: mobile check-in application for university campuses beyond COVID-19. | Yuuki Nishiyama, Hiroaki Murakami, Ryoto Suzuki, Kazusato Oko, Issey Sukeda, Kaoru Sezaki, Yoshihiro Kawahara |