Direct Distributional Optimization for Provable Alignment of Diffusion Models. - Ryotaro Kawata; Kazusato Oko; Atsushi Nitanda; Taiji Suzuki (ICLR, 2025)

When Does Metadata Conditioning (NOT) Work for Language Model Pre-Training? A Study with Context-Free Grammars. - Rei Higuchi; Ryotaro Kawata; Naoki Nishikawa; Kazusato Oko; Shoichiro Yamaguchi; Sosuke Kobayashi; Seiya Tokui; Kohei Hayashi; Daisuke Okanohara; Taiji Suzuki (CoRR, 2025)

Mixture of Experts Provably Detect and Learn the Latent Cluster Structure in Gradient-Based Learning. - Ryotaro Kawata; Kohsei Matsutani; Yuri Kinoshita; Naoki Nishikawa; Taiji Suzuki (CoRR, 2025)