Improve Knowledge Graph Completion for Diagnosing Defects in Main Electrical Equipment.

Jianye Huang,Jian Qian,Yanyu Chen,Rui Lin,Yuyou Weng,Guoqing Lin,Zhihong Zhang
DOI: https://doi.org/10.1007/978-981-99-4761-4_62
2023-01-01
Abstract:Defects in main electrical equipment can potentially cause power grid collapses, leading to catastrophic risks and immense pressure on dispatch operations. We adopt a Knowledge Graph Completion (KGC) approach to diagnose defects in main electrical equipment. The conventional approach for knowledge graph completion involves randomly initializing the feature representation of nodes and edges. In the context of main electrical equipment, we propose an innovative method based on ChatGPT to generate a corpus for fine-tuning the pre-trained model. We then employ the pre-trained model as a feature extractor for nodes and edges, enhancing the relationships between nodes and improving the initial embedding quality of both nodes and edges. Our pre-trained model fine-tuning process can be efficiently executed on a CPU with less than 10% of the parameters required for fine-tuning the entire pre-trained model. Our experimental results show that our model effectively improves the performance of KGC in diagnosing defects in main electrical equipment.
What problem does this paper attempt to address?