2 citations · 6 across the 11 of their papers we have counts for
4 papers · 1 filter
Expand and Prune: Maximizing Trajectory Diversity for Effective GRPO in Generative Models
Shiran Ge, Chenyi Huang, Yuang Ai +3
Group Relative Policy Optimization (GRPO) is a powerful technique for aligning generative models, but its effectiveness is bottlenecked by the conflict between large group sizes an…
Rectifying Magnitude Neglect in Linear Attention
Qihang Fan, Huaibo Huang, Yuang Ai +1
As the core operator of Transformers, Softmax Attention exhibits excellent global modeling capabilities. However, its quadratic complexity limits its applicability to vision tasks.…
Fast and Accurate Image Restoration and Generation with Rank Enhanced Linear Attention
Yuang Ai
Transformer-based models have made remarkable progress in image restoration (IR) tasks. However, the quadratic complexity of self-attention in Transformer hinders its applicability…
DiCo: Revitalizing ConvNets for Scalable and Efficient Diffusion Modeling
Yuang Ai, Qihang Fan, Xuefeng Hu +3
Diffusion Transformer (DiT), a promising diffusion model for visual generation, demonstrates impressive performance but incurs significant computational overhead. Intriguingly, ana…