A-SelecT: Automatic Timestep Selection for Diffusion Transformer Representation LearningThe paper presents A-SelecT, an innovative technique for automatic timestep selection in Diffusion Transformer (DiT) representation learning, aimed at enhancing training efficiency and representational capacity. A-SelecT identifies the most informative timesteps during a single run, effectively removing the need for extensive exhaustive searches. Experimental results indicate that DiT, augmented by A-SelecT, outperforms previous diffusion models in both classification and segmentation tasks. This advancement highlights its potential for improving discriminative tasks through enhanced generative pre-training.