Aug, 2024

不变图学习与信息瓶颈的结合用于分布外泛化

TL;DR本文针对图学习中分布外(OOD)泛化的主要挑战,提出了一种基于信息瓶颈理论的不变图学习框架(InfoIGL),旨在提取图的稳定特征并提高模型对未见分布的泛化能力。通过引入冗余过滤器和多层次对比学习,InfoIGL在不依赖不变性监督信号的情况下,显著提升了图分类任务的OOG泛化性能,实验结果显示该方法在合成和真实世界数据集上实现了最先进的性能。