H$^2$CL: Heterogeneity-Aware Hypergraph Contrastive Learning for Robust Representation Learning
Abstract
Lay Summary
Many real-world data sets are not just made of pairs of connected items. For example, a scientific paper may connect several authors, a purchase record may connect a user with several products, and a recommendation system may connect many related actions. Existing AI methods often assume that all items grouped together are similar, but this can be misleading when one item belongs to a different topic or community. Such mixed connections can pass confusing information through the model and reduce the quality of its predictions. Our work introduces H²CL, a method that first checks how well each item fits the group it belongs to. It then builds cleaner training examples by reducing the influence of confusing connections, and it also adjusts how much information each item is allowed to pass through a group. This helps the model learn from useful group relationships while ignoring misleading signals. Experiments on several standard and larger data sets show that this approach improves prediction accuracy and remains more reliable when extra noisy connections are added. Overall, H²CL provides a more robust way to learn from complex grouped data in applications such as citation analysis, recommendation systems, and other network-based AI tasks.