arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2203.12893cs.CV
收藏

FAMLP:一种用于领域泛化的频率感知MLP类架构

FAMLP: A Frequency-Aware MLP-Like Architecture For Domain Generalization

University of Science and Technology of China(中国科学技术大学)

Kecheng Zheng, Yang Cao, Kai Zhu, Ruijing Zhao, Zheng-Jun Zha

摘要:

完全建立在多层感知器上的MLP模型最近被重新审视,表现出与变压器相当的性能。它是最有前途的架构之一,由于在大规模识别任务的网络能力和效率之间的良好权衡。然而,其对异构任务的泛化性能不如其他架构(例如,CNN和transformers),这是由于域信息的广泛保留。为了解决这个问题,我们提出了一种新的频率感知的MLP架构,在该架构中,特定于域的特征在变换后的频域中被过滤掉,增加了用于标签预测的不变描述符。具体来说,我们设计了一个自适应傅立叶滤波器层,其中一个可学习的频率滤波器是用来调整幅度分布,通过优化的实部和虚部。一个低秩增强模块被进一步提出,通过添加SVD分解的低频分量来校正滤波后的特征。最后,利用动量更新策略,通过加权历史状态的输出蒸馏,使优化对模型参数和输入的波动具有稳定性。据我们所知,我们是第一个提出一个MLP样的骨干领域泛化。在三个基准上进行的大量实验显示出显着的泛化性能,分别比最先进的方法高出3%,4%和9%。

英文摘要:

MLP-like models built entirely upon multi-layer perceptrons have recently been revisited, exhibiting the comparable performance with transformers. It is one of most promising architectures due to the excellent trade-off between network capability and efficiency in the large-scale recognition tasks. However, its generalization performance to heterogeneous tasks is inferior to other architectures (e.g., CNNs and transformers) due to the extensive retention of domain information. To address this problem, we propose a novel frequency-aware MLP architecture, in which the domain-specific features are filtered out in the transformed frequency domain, augmenting the invariant descriptor for label prediction. Specifically, we design an adaptive Fourier filter layer, in which a learnable frequency filter is utilized to adjust the amplitude distribution by optimizing both the real and imaginary parts. A low-rank enhancement module is further proposed to rectify the filtered features by adding the low-frequency components from SVD decomposition. Finally, a momentum update strategy is utilized to stabilize the optimization to fluctuation of model parameters and inputs by the output distillation with weighted historical states. To our best knowledge, we are the first to propose a MLP-like backbone for domain generalization. Extensive experiments on three benchmarks demonstrate significant generalization performance, outperforming the state-of-the-art methods by a margin of 3%, 4% and 9%, respectively.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑