An Enhanced Knowledge Graph Embedding for Small-Scale Sparse Knowledge Graph
Yushun Xie, Zhaoquan Gu, Haiyan Wang, Runnan Tan, Xiangyu Song
Lecture notes in computer science
Problems Identified (4)
Poor KGE on small sparse graphs: Existing knowledge graph embedding techniques perform poorly on small-scale datasets because small knowledge graphs are incomplete and sparse, leading to overfitting.
Temporal dependency preservation in attack KGs: Applying KGE to the Cybersecurity Attack Knowledge Graph requires handling temporal attributes to preserve temporal dependencies of attack steps.
Poor KGE on small sparse graphs: Existing knowledge graph embedding techniques perform poorly on small-scale datasets because small knowledge graphs are incomplete and sparse, leading to overfitting.
Temporal dependency preservation in attack KGs: Applying KGE to the Cybersecurity Attack Knowledge Graph requires handling temporal attributes to preserve temporal dependencies of attack steps.
Proposed Solutions (5)
Enhanced small-graph KGE: The paper proposes an enhanced knowledge graph embedding method for small-scale knowledge graphs that combines rule-based data augmentation with neighborhood-based embedding enhancement.
Temporal attribute abstraction: The method abstracts temporal attributes into knowledge to preserve temporal dependencies in cybersecurity attack steps.
Spatial-rule ranking correction: The method corrects KGE ranking by using spatial rule scores.
Enhanced small-graph KGE: The paper proposes an enhanced knowledge graph embedding method for small-scale knowledge graphs that combines rule-based data augmentation with neighborhood-based embedding enhancement.
Temporal attribute abstraction: The method abstracts temporal attributes into knowledge to preserve temporal dependencies in cybersecurity attack steps.
Results (2)
Improved KGE link prediction metrics:
Improved KGE link prediction metrics:
Research Domain
Knowledge graph embedding for small-scale sparse knowledge graphs and cybersecurity attack knowledge graphs