Zihao He

Bend the Basics: Degradation-Aware Deformable Tokenization for All-in-One Image Restoration

FIT employs a lightweight Degradation Encoder to predict a global degradation vector and a spatial degradation map from local degradation severity, which jointly condition the patch embedding and unembedding through adaptive deformation, and introduces a task-token dropout strategy that regularizes task conditioning during training.