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

通过双曲状态空间幻觉学习细粒度域泛化

Learning Fine-grained Domain Generalization via Hyperbolic State Space Hallucination

School of Artificial Intelligence, Wuhan University(武汉大学人工智能学院);Faculty of Information Technology, Monash University(莫纳什大学信息技术学院);School of Medicine, Yale University(耶鲁大学医学院)

Qi Bi, Jingjun Yi, Haolan Zhan, Wei Ji, Gui-Song Xia

AI总结:

本文提出双曲状态空间幻觉方法,通过风格幻觉丰富状态嵌入多样性并利用双曲流形建模高阶统计量,以消除风格变化对细粒度模式的影响,在三个细粒度域泛化基准上取得最优性能。

摘要:

细粒度域泛化(FGDG)旨在仅使用源域数据训练时,学习一种能够良好泛化到未见目标域的细粒度表示。与通用域泛化相比,FGDG 尤其具有挑战性,因为细粒度类别只能通过一些细微且微小的模式加以区分。这些模式在由光照、颜色等引起的跨域风格变化下尤为脆弱。为推进这一前沿方向,本文提出了一种新颖的双曲状态空间幻觉(Hyperbolic State Space Hallucination,HSSH)方法。该方法包含两个关键组件,即状态空间幻觉(state space hallucination,SSH)和双曲流形一致性(hyperbolic manifold consistency,HMC)。SSH 通过首先对源图像进行外推、然后进行幻觉化,丰富状态嵌入的风格多样性。随后,将风格幻觉前后的状态嵌入投影到双曲流形中。双曲状态空间对高阶统计量进行建模,从而能够更好地区分细粒度模式。最后,最小化双曲距离,以消除风格变化对细粒度模式的影响。在三个 FGDG 基准上的实验证明了该方法的当前最优性能。

英文摘要:

Fine-grained domain generalization (FGDG) aims to learn a fine-grained representation that can be well generalized to unseen target domains when only trained on the source domain data. Compared with generic domain generalization, FGDG is particularly challenging in that the fine-grained category can be only discerned by some subtle and tiny patterns. Such patterns are particularly fragile under the cross-domain style shifts caused by illumination, color and etc. To push this frontier, this paper presents a novel Hyperbolic State Space Hallucination (HSSH) method. It consists of two key components, namely, state space hallucination (SSH) and hyperbolic manifold consistency (HMC). SSH enriches the style diversity for the state embeddings by firstly extrapolating and then hallucinating the source images. Then, the pre- and post- style hallucinate state embeddings are projected into the hyperbolic manifold. The hyperbolic state space models the high-order statistics, and allows a better discernment of the fine-grained patterns. Finally, the hyperbolic distance is minimized, so that the impact of style variation on fine-grained patterns can be eliminated. Experiments on three FGDG benchmarks demonstrate its state-of-the-art performance.

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

↑