Can We Change "Stroke Size" to Improve Flow Matching?
Yunwei Bai, Ying Kiat Tan, Tsuhan Chen
原文摘要
Deep-learning-based image generation models are useful but difficult to train. Inspired by the varying stroke sizes used in human painting, we hypothesize that applying ``rough strokes'' can attenuate prediction biases and improve flow-matching image generation. We develop MultiStroke, which progressively varies the spatial granularity of the clean-signal component in the velocity regression target. A lightweight corrector aligns adjacent roughness levels during sampling. On ImageNet-1K, MultiStroke reduces FID by $22.5\%$ relative to a matched flow-matching baseline, despite the limited corrector capacity. Code is available at https://github.com/DaChengRuoQue/MultiStroke-Code. A quick video explanation is available at https://dachengruoque.github.io/MultiStroke-Web/.
cs.CV · cs.AI