Chen-Hao Chao

Ph.D. in CS @ University of Toronto

chc2.png
Hello, I'm a Computer Science Ph.D. student at the University of Toronto (UofT), advised by Prof. Rahul G. Krishnan. I am honored to be a recipient of the Mary H. Beatty Fellowship. I completed my master's and bachelor's degrees in Computer Science at National Tsing Hua University (NTHU). I collaborated on research projects with Prof. Chun-Yi Lee, visited Prof. Zsolt Kira's lab at Georgia Tech, and interned at NVIDIA and MediaTek.

My current research explores efficient pre-training methods and scaling behavior of diffusion language models. More broadly, my works cover probabilistic modeling and generative AI, spanning both discrete and continuous generative methods (e.g., score-based and flow-based models) with applications to reinforcement learning, visual domain adaptation, and biological data visualization. I built:
  • MDM-Prime (v1, v2): A scalable diffusion language model that allows partial word editings.
  • EBFlow/MEow: A flow-based model that enables energy-based reinterpretation and RL applications.
  • QCSBM: An architecturally flexible diffusion model that satisfies the conservative property.
  • DLSM: A classifier-guidance diffusion model enhanced by likelihood score matching.

latest posts [full list]

selected publications [full list]

  1. EMNLP
    mdmprime-v2.png
    MDM-Prime-v2: Binary Encoding and Index Shuffling Enable Scaling of Diffusion Language Models
    Chen-Hao Chao, Wei-Fang Sun, Junwei Quan, and 2 more authors
    Proceedings of the Conference on Empirical Methods in Natural Language Processing (EMNLP), 2026
  2. NeurIPS
    mdmprime.png
    Beyond Masked and Unmasked: Discrete Diffusion Models via Partial Masking
    Chen-Hao Chao, Wei-Fang Sun, Hanwen Liang, and 2 more authors
    In Proceedings of the Conference on Neural Information Processing Systems (NeurIPS) , 2025
  3. NeurIPS
    meow.png
    Maximum Entropy Reinforcement Learning via Energy-Based Normalizing Flow
    Chen-Hao Chao*, Chien Feng*, Wei-Fang Sun, and 3 more authors
    In Proceedings of the Conference on Neural Information Processing Systems (NeurIPS) , 2024
  4. NeurIPS
    ebflow.png
    Training Energy-Based Normalizing Flow with Score-Matching Objectives
    Chen-Hao Chao, Wei-Fang Sun, Yen-Chang Hsu, and 2 more authors
    In Proceedings of the Conference on Neural Information Processing Systems (NeurIPS) , 2023
  5. ICML
    qcsbm.png
    On Investigating the Conservative Property of Score-Based Generative Models
    Chen-Hao Chao, Wei-Fang Sun, Bo-Wun Cheng, and 1 more author
    In Proceedings of the International Conference on Machine Learning (ICML) , 2023
  6. ICLR
    dlsm.png
    Denoising Likelihood Score Matching for Conditional Score-based Data Generation
    Chen-Hao Chao, Wei-Fang Sun, Bo-Wun Cheng, and 6 more authors
    In Proceedings of the International Conference on Learning Representations (ICLR) , 2022
  7. TPAMI
    rainbow.png
    Rainbow UDA: Combining Domain Adaptive Models for Semantic Segmentation Tasks
    Chen-Hao Chao, Bo-Wun Cheng, Tzu-Wen Wang*, and 2 more authors
    IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2023