Freqformer: Image-Demoir\'eing Transformer via Effective Frequency Decomposition
Summary
Freqformer is a novel Transformer-based framework for image demoiréing, addressing the complex interplay of texture corruption and color distortions caused by moiré patterns. Developed by researchers from Shanghai Jiao Tong and Harvard Universities, it employs an effective frequency decomposition to separate moiré into high-frequency textures and low-frequency color distortions. A dual-branch architecture, with crop-based training for high-frequency and resize-based training for low-frequency components, processes these distinct characteristics. Freqformer also introduces a learnable Frequency Composition Transform (FCT) for adaptive output fusion and a Spatial-Aware Channel Attention (SA-CA) module for efficient feature refinement. Extensive experiments on FHDMi and UHDM benchmarks demonstrate Freqformer achieves state-of-the-art performance with a compact model size of 6.065M parameters and 2.49 TFLOPS for 4K images.
Key takeaway
For Machine Learning Engineers developing image processing solutions, Freqformer offers a highly efficient and effective approach to image demoiréing. Its novel frequency decomposition and dual-branch architecture, combined with a learnable Frequency Composition Transform, enable superior artifact removal and color restoration, even for 4K images. You should consider integrating this compact 6.065M parameter model for high-resolution applications or edge device deployment where computational efficiency is critical.
Key insights
Freqformer disentangles moiré patterns into distinct frequency components for targeted, efficient image restoration.
Principles
- Moiré patterns separate into high-frequency textures and low-frequency color distortions.
- High-frequency components benefit from crop-based training due to spatial locality.
- Low-frequency components are scale-robust, enabling resize-based training.
Method
Freqformer uses a two-stage training: independent high/low-frequency branch training, followed by joint fine-tuning with a learnable FCT. It employs a recursive convolution-based frequency decomposition.
In practice
- Deploy on edge devices due to compact model size (6.065M parameters).
- Process ultra-high-resolution images (e.g., 4K) efficiently (2.49 TFLOPS).
- Improve demoiréing by separating texture and color artifacts.
Topics
- Image Demoiréing
- Transformer Networks
- Frequency Decomposition
- Dual-Branch Architecture
- Spatial-Aware Channel Attention
- High-Resolution Imaging
Code references
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.