Skip to content
/ ca2kg Public

Accompanying repository for the paper β€œWhen to Trust: Causality-Aware Calibration for Accurate KG-RAG” by Jing Ren, Bowen Li, Ziqi Xu, Xikun Zhang, Haytham Fayek, and Xiaodong Li, published in The Web Conference 2026.

License

Notifications You must be signed in to change notification settings

Aisuko/ca2kg

Repository files navigation

Ca2KG

Ca2KG

Python Jupyter License


🎯 Framework

Framework


πŸ—οΈ Repository Structure

kg-rag/
β”œβ”€β”€ πŸ“ imgs/                     # Figures and visualizations
β”‚   β”œβ”€β”€ fig_accuracy_under_token_caps_metaqa.png
β”‚   └── fig_answer_token_efficiency_metaqa_committed.png
β”œβ”€β”€ πŸ“ methods/                 # RAG implementation notebooks
β”œβ”€β”€ πŸ“ helper/                  # Utility classes and tools
β”œβ”€β”€ πŸ“ ablation/               # Ablation study experiments
β”œβ”€β”€ πŸ“ efficiency_experiment/   # Efficiency analysis notebooks
└── πŸ“„ README.md

πŸš€ Quick Start

Installation & Setup
# Note: All code in this repository has been successfully tested and executed on Kaggle platform.
# Clone the repository
cd kg-rag

# Launch Jupyter notebooks
jupyter notebook

πŸ”¬ Experiments

Component Description Location
Core Methods RAG implementation notebooks methods/
Ablation Studies Component analysis experiments ablation/
Efficiency Analysis Performance optimization studies efficiency_experiment/
Helper Tools Utility classes and functions helper/

πŸ“ˆ Contributions

  • βœ… Lower Calibration Error: Improved confidence calibration for knowledge-based QA
  • βœ… Higher Accuracy: Enhanced performance on MetaQA and benchmark datasets
  • βœ… Efficient Reasoning: Optimized multi-hop reasoning with reduced computational overhead
  • βœ… Comprehensive Evaluation: Extensive ablation studies and efficiency analysis

πŸ“š Citation

See CITATION.cff for the preferred citation metadata.


Built with ❀️ for advancing knowledge-grounded AI research

About

Accompanying repository for the paper β€œWhen to Trust: Causality-Aware Calibration for Accurate KG-RAG” by Jing Ren, Bowen Li, Ziqi Xu, Xikun Zhang, Haytham Fayek, and Xiaodong Li, published in The Web Conference 2026.

Topics

Resources

License

Stars

Watchers

Forks