Kohsei Matsutani

I am a 1st year master student at the University of Tokyo under the supervision of Prof. Yutaka Matsuo. My research interests center on machine learning, currently focusing on the reasoning capabilities of Large Language Models (LLMs). I am currently interning at Liquid AI.

Email  /  Scholar  /  Linkedin

profile photo

Research

OrderGrad
OrderGrad: Optimizing Beyond the Mean with Order-Statistic Policy Gradient Estimation
Paavo Parmas, Yongmin Kim, Kohsei Matsutani, Shota Takashiro, Soichiro Nishimori, Takeshi Kojima, Yusuke Iwasawa, Yutaka Matsuo,
Preprint, 2026
arXiv

MaxPO
On Advantage Estimates for Max@K Policy Gradients
Shota Takashiro, Soichiro Nishimori, Paavo Parmas, Yongmin Kim, Kohsei Matsutani, Gouki Minegishi, Yusuke Iwasawa, Takeshi Kojima, Yutaka Matsuo,
Preprint, 2026
arXiv

Zipping
Zipping the Thought: When and How Compressed Reasoning Data Works in LLM Post-Training
Kohsei Matsutani, Gouki Minegishi, Takeshi Kojima, Yusuke Iwasawa, Yutaka Matsuo,
Preprint, 2026
arXiv

MultiBanana
MultiBanana: A Challenging Benchmark for Multi-Reference Text-to-Image Generation
Yuta Oshima, Daiki Miyake, Kohsei Matsutani, Yusuke Iwasawa, Masahiro Suzuki, Yutaka Matsuo, Hiroki Furuta
CVPR, 2026
arXiv

RL Squeezes SFT Expands Image
RL Squeezes, SFT Expands: A Comparative Study of Reasoning LLMs
Kohsei Matsutani, Shota Takashiro, Gouki Minegishi, Takesi Kojima, Yusuke Iwasawa, Yutaka Matsuo
ICLR, 2026
arXiv

MoE Provably Image
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
* Equal contribution
ICML, 2025
arXiv