ToMA: Token Merge with Attention for Diffusion Models

Wenbo Lu, Shaoyi Zheng, Yuxuan Xia, Shengjie Wang. ToMA: Token Merge with Attention for Diffusion Models. In Forty-second International Conference on Machine Learning, ICML 2025, Vancouver, BC, Canada, July 13-19, 2025. OpenReview.net, 2025. [doi]

Authors

Wenbo Lu

This author has not been identified. Look up 'Wenbo Lu' in Google

Shaoyi Zheng

This author has not been identified. Look up 'Shaoyi Zheng' in Google

Yuxuan Xia

This author has not been identified. Look up 'Yuxuan Xia' in Google

Shengjie Wang

This author has not been identified. Look up 'Shengjie Wang' in Google