arXiv:2410.22026cs.LGcs.AI2024-10

通过二阶池化提升双曲表示学习的层次区分能力。

Enhance Hyperbolic Representation Learning via Second-order Pooling

  • 引入二阶池化增强不同层级样本间的距离。
  • 新方法使骨干网络无需高Lipschitz常数仍保持良好泛化。
  • 适合处理图结构数据的层次表示学习任务。

双曲表示学习擅长捕捉层次结构信息,但不同层级类别样本间的距离往往需较大。我们发现,双曲判别目标迫使骨干网络捕捉这种层次信息,可能不可避免地增大其Lipschitz常数,从而限制骨干网络的泛化能力。为解决此问题,我们引入二阶池化,可自然增大样本间距离,且不损害输入特征的泛化能力,因此骨干网络的Lipschitz常数不必过大。然而,现有低维双线性池化方法无法直接用于双曲表示学习,因其会削弱距离扩展能力。为此,我们提出核近似正则化,使低维双线性特征在低维空间中能良好逼近核函数。我们在图结构数据集上进行了大量实验,验证了该方法的有效性。

原文摘要 · Abstract (English)

Hyperbolic representation learning is well known for its ability to capture hierarchical information. However, the distance between samples from different levels of hierarchical classes can be required large. We reveal that the hyperbolic discriminant objective forces the backbone to capture this hierarchical information, which may inevitably increase the Lipschitz constant of the backbone. This can hinder the full utilization of the backbone's generalization ability. To address this issue, we introduce second-order pooling into hyperbolic representation learning, as it naturally increases the distance between samples without compromising the generalization ability of the input features. In this way, the Lipschitz constant of the backbone does not necessarily need to be large. However, current off-the-shelf low-dimensional bilinear pooling methods cannot be directly employed in hyperbolic representation learning because they inevitably reduce the distance expansion capability. To solve this problem, we propose a kernel approximation regularization, which enables the low-dimensional bilinear features to approximate the kernel function well in low-dimensional space. Finally, we conduct extensive experiments on graph-structured datasets to demonstrate the effectiveness of the proposed method.

双曲学习层次表示池化方法

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。