TGHE: Template-based Graph Homomorphic Encryption for Privacy-Preserving GNN Inference in Edge-Cloud Systems
Summary
TGHE (Template-based Graph Homomorphic Encryption) is a novel ego-centric framework designed for privacy-preserving Graph Neural Network (GNN) inference in edge-cloud systems. It addresses the limitations of existing graph-centric homomorphic encryption (HE)-based GNN systems, which tie per-query cost to global graph size and are restricted to graphs of at most ~20k nodes, making them impractical for large-scale dynamic financial graphs. TGHE resolves this by leveraging a "template phenomenon" where local computation trees in transaction graphs converge into a small set of structural shapes. The framework canonicalizes ego-graphs at the edge and packs structurally identical trees into shared CKKS ciphertexts, enabling SIMD-parallel encrypted inference. It incorporates two optimizers, Approximate Template Fitting and Topology Collapse, to ensure comprehensive SIMD coverage. On the DGraphFin dataset, which contains 3.7M nodes and 4.3M edges, TGHE-Collapse achieved a 66.9x speedup compared to the sequential encrypted baseline, with a minimal AUC loss of less than 0.002.
Key takeaway
For AI Security Engineers designing privacy-preserving GNN systems for large-scale dynamic graphs, TGHE offers a critical paradigm shift. You should consider adopting an ego-centric, template-based approach to overcome the scalability limitations of traditional graph-centric homomorphic encryption. This enables efficient SIMD-parallel encrypted inference on millions of nodes, significantly boosting performance and preserving privacy for sensitive data like financial transactions.
Key insights
Ego-centric graph processing with template-based packing enables efficient, privacy-preserving GNN inference on large dynamic graphs.
Principles
- Graph computation trees often share structural templates.
- Packing identical structures enables SIMD-parallel HE.
- Ego-centric processing scales better for large graphs.
Method
TGHE canonicalizes ego-graphs at the edge, then packs structurally identical trees into shared CKKS ciphertexts for SIMD-parallel encrypted inference, using Approximate Template Fitting and Topology Collapse for full SIMD coverage.
In practice
- Use ego-centric processing for large dynamic graphs.
- Identify structural templates in graph computations.
- Apply SIMD-parallel HE for encrypted inference.
Topics
- Homomorphic Encryption
- Graph Neural Networks
- Privacy-Preserving AI
- Edge-Cloud Systems
- SIMD Parallelism
- Financial Graphs
- CKKS Ciphertext
Best for: Research Scientist, AI Architect, AI Engineer, AI Scientist, Machine Learning Engineer, AI Security Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.