CORTEXA
← Browse

Chengru Song

2 papers indexed

arxivcs.CVcs.LG2026-07-22

HeadCast: Casting Attention Heads for Efficient Autoregressive Video Generation

Jinliang Shen, Lianghao Su, Zheming Li, Kang He, ZiLiang Lai, Yanbing Jiang, et al.

Autoregressive (AR) video diffusion models have become a promising paradigm for long and streaming video synthesis, but the continuously growing Key-Value (KV) cache makes attention the dominant inference cost, especially at high resolution where each frame contributes many token…

View free PDFSource page
arxivcs.LGcs.CVeess.SY2026-07-20

JAGG: Jacobian-Aggregated Group Gradient for Efficient GRPO Training of Diffusion Models

Ruiyi Ding, Jie Li, He Kang, Ziyan Liu, Chengru Song, Yuan chen

Group Relative Policy Optimization (GRPO) is a powerful reinforcement learning algorithm for aligning generative models with human preferences. While successful in large language models~\cite{shao2024deepseekmathpushinglimitsmathematical}, its extension to diffusion and flow matc…

View free PDFSource page