Nothing Special   »   [go: up one dir, main page]

skip to main content
research-article

Feature enhancement based on hierarchical reconstruction framework for inductive prediction on sparse graphs

Published: 20 February 2025 Publication History

Abstract

Knowledge graph completion aims to infer the missing links of new elements, however, the missing links often lie in sparse regions of the graph. Primary subgraph-based methods rely heavily on structural information, which makes it difficult for them to play an essential role in sparse graph completion. To address this challenge, we propose a learning framework for feature-enhanced hierarchical reconstruction (FEHR). The proposed FEHR explores relational semantics at the global and local levels, minimizing the limitations of sparse structures. First, entity graphs are converted into relation graphs, and overreliance on the entity structure is reduced by obtaining prior knowledge on similar global graphs. Second, the relational features are further refined at the local level. Finally, an improved performer model expresses the degree of preference between the predicted behaviors and relations. Extensive inductive experiments showed that FEHR performs better than state-of-the-art baselines, achieving improvements in area under the prediction–recall curve (AUC-PR) and Hits@n metrics, ranging from 0.32% to 11.73%.

Highlights

We construct a novel hierarchical reconstructed feature learning framework.
Global acquisition of additional prior knowledge through similarity graphs.
Locally distinguish the degree of association of relationships.
FEHR alleviates the limitation of a few links in sparse graphs.
Our model outperforms advanced baselines on most benchmark datasets.

References

[1]
Aleksandar, P., & Emanuel, S. (2023). ExpressivE: A Spatio-Functional Embedding For Knowledge Graph Completion. In The eleventh international conference on learning representations (pp. 1–45). Kigali, Rwanda, Rwanda: https://doi.org/10.34726/5422.
[2]
Alhamoud K., Ghunaim Y., Alshehri A.S., Li G., Ghanem B., You F., Leveraging 2D molecular graph pretraining for improved 3D conformer generation with graph neural networks, Computers & Chemical Engineering 183 (2024),.
[3]
Auer S., Bizer C., Kobilarov G., Dbpedia: A nucleus for a web of open data, in: The semantic web, 2007, pp. 722–735,.
[4]
Baek J., Lee D.B., Hwang S.J., Learning to extrapolate knowledge: Transductive few-shot out-of-graph link prediction, in: Advances in neural information processing systems, vol. 33, Curran Associates, Inc., 2020, pp. 546–560. URL http://hdl.handle.net/10203/290460.
[5]
Boonyarat P., Liew D.J., Chang Y.-C., Leveraging enhanced BERT models for detecting suicidal ideation in thai social media content amidst COVID-19, Information Processing & Management 61 (2024),.
[6]
Bordes A., Usunier N., Garcia-Duran A., Weston J., Yakhnenko O., Translating embeddings for modeling multi-relational data, in: Burges C., Bottou L., Welling M., Ghahramani Z., Weinberger K. (Eds.), Advances in neural information processing systems, vol. 26, Curran Associates, Inc., 2013, doi:https://hal.science/hal-00920777.
[7]
Cao Z., Xu Q., Yang Z., Cao X., Huang Q., Dual quaternion knowledge graph embeddings, Proceedings of the AAAI Conference on Artificial Intelligence 35 (8) (2021) 6894–6902,.
[8]
Chao L., He J., Wang T., Chu W., PairRE: Knowledge graph embeddings via paired relation vectors, in: Proceedings of the 59th annual meeting of the association for computational linguistics and the 11th international joint conference on natural language processing, 2021, pp. 4360–4369,.
[9]
Chen J., He H., Wu F., Wang J., Topology-aware correlations between relations for inductive link prediction in knowledge graphs, Proceedings of the AAAI Conference on Artificial Intelligence 35 (7) (2021) 6271–6278,.
[10]
Chen M., Zhang W., Zhu Y., Zhou H., Yuan Z., Xu C., Chen H., Meta-knowledge transfer for inductive knowledge graph embedding, in: Proceedings of the 45th international ACM SIGIR conference on research and development in information retrieval, Association for Computing Machinery, New York, NY, USA, 2022, pp. 927–937,.
[11]
Choromanski K.M., Likhosherstov V., Dohan D., Song X., Gane A., Sarlos T., Hawkins P., Davis J.Q., Mohiuddin A., Kaiser L., Belanger D.B., Colwell L.J., Weller A., Rethinking attention with performers, in: International conference on learning representations, 2021, pp. 1–38. doi:https://openreview.net/forum?id=Ua6zuk0WRH.
[12]
Deng W., Zhang Y., Yu H., Li H., Knowledge graph embedding based on dynamic adaptive atrous convolution and attention mechanism for link prediction, Information Processing & Management 61 (3) (2024),.
[13]
Dettmers T., Minervini P., Stenetorp P., Riedel S., Convolutional 2D knowledge graph embeddings, Proceedings of the AAAI Conference on Artificial Intelligence 32 (1) (2018),.
[14]
Duan Y., Tang J., Xu H., Liu C., Zeng W., Commonsense-guided inductive relation prediction with dual attention mechanism, Applied Sciences 14 (5) (2024),.
[15]
Feng J.R., Zhao M., Yu G., Zhang J., Lu S., Dynamic risk analysis of accidents chain and system protection strategy based on complex network and node structure importance, Reliability Engineering & System Safety 238 (2023),.
[16]
Finzi M., Welling M., Wilson A.G.G., A practical method for constructing equivariant multilayer perceptrons for arbitrary matrix groups, in: Proceedings of the 38th international conference on machine learning, vol. 139, PMLR, 2021, pp. 3318–3328. URL https://proceedings.mlr.press/v139/finzi21a.html.
[17]
Ju W., Yang J., Qu M., Song W., Shen J., Zhang M., KGNN: Harnessing kernel-based networks for semi-supervised graph classification, in: Proceedings of the fifteenth ACM international conference on web search and data mining, Association for Computing Machinery, New York, NY, USA, 2022, pp. 421–429,.
[18]
Lee J., Chung C., Whang J.J., Ingram: Inductive knowledge graph embedding via relation graphs, in: Proceedings of the 40th international conference on machine learning, vol. 202, ML Research Press, 2023, pp. 18796–18809. URL http://hdl.handle.net/10203/314508.
[19]
Li J., Wang Q., Mao Z., Inductive relation prediction from relational paths and context with hierarchical transformers, in: 2023 IEEE international conference on acoustics, speech and signal processing, 2023, pp. 1–5,.
[20]
Li L., Wang W., Yang Y., LogicSeg: Parsing visual semantics with neural logic learning and reasoning, in: Proceedings of the IEEE/CVF international conference on computer vision, 2023, pp. 4122–4133,.
[21]
Liang K., Meng L., Zhou S., Tu W., Wang S., Liu Y., Liu M., Zhao L., Dong X., Liu X., MINES: Message intercommunication for inductive relation reasoning over neighbor-enhanced subgraphs, Proceedings of the AAAI Conference on Artificial Intelligence 38 (9) (2024) 10645–10653,.
[22]
Lin Z., Xie J., Li Q., Multi-modal news event detection with external knowledge, Information Processing & Management 61 (3) (2024),.
[23]
Liu J., Chabot Y., Troncy R., Huynh V.-P., Labbé T., Monnin P., From tabular data to knowledge graphs: A survey of semantic table interpretation tasks and methods, Journal of Web Semantics 76 (2023),.
[24]
Liu S., Wang X., Macdonald C., Ounis I., A social-aware Gaussian pre-trained model for effective cold-start recommendation, Information Processing & Management 61 (2) (2024),.
[25]
Luo L., Haffari G., Pan S., Graph sequential neural ODE process for link prediction on dynamic and sparse graphs, in: Proceedings of the sixteenth ACM international conference on web search and data mining, Association for Computing Machinery, New York, NY, USA, 2023, pp. 778–786,.
[26]
Mai S., Zheng S., Yang Y., Hu H., Communicative message passing for inductive relation reasoning, Proceedings of the AAAI Conference on Artificial Intelligence 35 (5) (2021) 4294–4302,.
[27]
Meilicke C., Fink M., Wang Y., Ruffinelli D., Gemulla R., Stuckenschmidt H., Fine-grained evaluation of rule-and embedding-based systems for knowledge graph completion, in: The semantic web – ISWC 2018, Springer, Cham, 2018, pp. 3–20,.
[28]
Mikhail G., Etienne D., Jiapeng W., NodePiece: Compositional and parameter-efficient representations of large knowledge graphs, in: International conference on learning representations, 2022, pp. 1–25. URL https://openreview.net/forum?id=xMJWUKJnFSw.
[29]
Mohamed H.A., Pilutti D., James S., Del Bue A., Pelillo M., Vascon S., Locality-aware subgraphs for inductive link prediction in knowledge graphs, Pattern Recognition Letters 167 (2023) 90–97,.
[30]
Sadeghian A., Armandpour M., Ding P., Wang D.Z., DRUM: end-to-end differentiable rule mining on knowledge graphs, in: Proceedings of the 33rd international conference on neural information processing systems, vol. 32, Curran Associates Inc., Red Hook, NY, USA, 2019, pp. 15347–15357. URL https://dl.acm.org/doi/abs/10.5555/3454287.3455662.
[31]
Shen, J., Wang, C., Gong, L., & Song, D. (2022). Joint Language Semantic and Structure Embedding for Knowledge Graph Completion. In Proceedings of the 29th international conference on computational linguistics (pp. 1965–1978). Gyeongju, Republic of Korea: URL.
[32]
Sun Z., Deng Z.-H., Nie J.-Y., Tang J., Rotate: Knowledge graph embedding by relational rotation in complex space, in: International conference on learning representations, 2019, pp. 1–18. URL https://openreview.net/forum?id=HkgEQnRqYQ.
[33]
Teru K., Denis E., Hamilton W., Inductive relation prediction by subgraph reasoning, in: Proceedings of the 37th international conference on machine learning, vol. 119, PMLR, 2020, pp. 9448–9457. doi:https://dl.acm.org/doi/abs/10.5555/3524938.3525814.
[34]
Toutanova K., Chen D., Pantel P., Poon H., Choudhury P., Gamon M., Representing text for joint embedding of text and knowledge bases, in: Proceedings of the 2015 conference on empirical methods in natural language processing, Association for Computational Linguistics, Lisbon, Portugal, 2015, pp. 1499–1509,.
[35]
Trouillon T., Welbl J., Riedel S., Gaussier E., Bouchard G., Complex embeddings for simple link prediction, in: Proceedings of the 33rd international conference on machine learning, in: Proceedings of machine learning research, vol. 48, PMLR, New York, New York, USA, 2016, pp. 2071–2080. doi:https://dl.acm.org/doi/abs/10.5555/3045390.3045609.
[36]
Vrandečić D., Wikidata: a new platform for collaborative data collection, in: Proceedings of the 21st international conference on world wide web, Association for Computing Machinery, New York, NY, USA, 2012, pp. 1063–1064,.
[37]
Wang H., Mi J., Guo X., Hu P., Meta-learning adaptation network for few-shot link prediction in heterogeneous social networks, Information Processing & Management 60 (5) (2023),.
[38]
Wang L., Zhao M., Ji H., Jiang Z., Li R., Hu Z., Lu X., Dialogue summarization enhanced response generation for multi-domain task-oriented dialogue systems, Information Processing & Management 61 (3) (2024),.
[39]
Xie S., Pan Q., Wang X., Luo X., Sugumaran V., Combining prompt learning with contextual semantics for inductive relation prediction, Expert Systems with Applications 238 (2024),.
[40]
Xiong W., Hoang T., Wang W.Y., DeepPath: A reinforcement learning method for knowledge graph reasoning, in: Proceedings of the 2017 conference on empirical methods in natural language processing, Association for Computational Linguistics, Copenhagen, Denmark, 2017, pp. 564–573,.
[41]
Xu K., Hu W., Leskovec J., Jegelka S., How powerful are graph neural networks?, in: International conference on learning representations, 2019, pp. 1–17. URL https://openreview.net/forum?id=ryGs6iA5Km.
[42]
Yang F., Yang Z., Cohen W.W., Differentiable learning of logical rules for knowledge base reasoning, in: Advances in neural information processing systems, vol. 30, Curran Associates, Inc., 2017, URL https://dl.acm.org/doi/abs/10.5555/3294771.3294992.
[43]
Yang B., Yih S.W.-t., He X., Gao J., Deng L., Embedding entities and relations for learning and inference in knowledge bases, in: Proceedings of the international conference on learning representations, 2015, pp. 1–12. URL https://www.microsoft.com/en-us/research/publication/embedding-entities-and-relations-for-learning-and-inference-in-knowledge-bases/.
[44]
Z H., V G.-B., Z X., Type-aware embeddings for multi-hop reasoning over knowledge graphs, in: 31st international joint conference on artificial intelligence, International Joint Conferences on Artificial Intelligence Organization, 2022, pp. 3078–3084. URL https://orca.cardiff.ac.uk/id/eprint/149582.
[45]
Zhang Z., Cai J., Zhang Y., Wang J., Learning hierarchy-aware knowledge graph embeddings for link prediction, Proceedings of the AAAI Conference on Artificial Intelligence 34 (03) (2020) 3065–3072,.
[46]
Zhang J., Li X., Tian J., Luo H., An integrated multi-head dual sparse self-attention network for remaining useful life prediction, Reliability Engineering & System Safety 233 (2023),.
[47]
Zhang Y., Yao Q., Knowledge graph reasoning with relational digraph, in: Proceedings of the ACM web conference 2022, Association for Computing Machinery, New York, NY, USA, 2022, pp. 912–924,.
[48]
Zhang X., Zhang C., Guo J., Peng C., Niu Z., Wu X., Graph attention network with dynamic representation of relations for knowledge graph completion, Expert Systems with Applications 219 (2023),.
[49]
Zheng S., Mai S., Sun Y., Hu H., Yang Y., Subgraph-aware few-shot inductive link prediction via meta-learning, IEEE Transactions on Knowledge and Data Engineering 35 (6) (2023) 6512–6517,.
[50]
Zhu Y., Zhou Y., Liu Y., Wang X., Li J., SLGNN: synthetic lethality prediction in human cancers based on factor-aware knowledge graph neural network, Bioinformatics 39 (2) (2023),.

Recommendations

Comments

Please enable JavaScript to view thecomments powered by Disqus.

Information & Contributors

Information

Published In

cover image Information Processing and Management: an International Journal
Information Processing and Management: an International Journal  Volume 62, Issue 1
Jan 2025
1582 pages

Publisher

Pergamon Press, Inc.

United States

Publication History

Published: 20 February 2025

Author Tags

  1. Inductive prediction
  2. Sparse graphs
  3. Relational features
  4. Preference degree

Qualifiers

  • Research-article

Contributors

Other Metrics

Bibliometrics & Citations

Bibliometrics

Article Metrics

  • 0
    Total Citations
  • 0
    Total Downloads
  • Downloads (Last 12 months)0
  • Downloads (Last 6 weeks)0
Reflects downloads up to 19 Feb 2025

Other Metrics

Citations

View Options

View options

Figures

Tables

Media

Share

Share

Share this Publication link

Share on social media