Privately Learning from Graphs with Applications in Fine-tuning Large Language Models

Haoteng Yin, Rongzhe Wei, Eli Chien, Pan Li
10/10/2024
cs.LGcs.CLcs.CR

Official code: Graph-COM/PvGaLM (3 stars, Apache-2.0, last commit 2024-10-15).

Abstract

Graphs offer unique insights into relationships between entities, complementing data modalities like text and images and enabling AI models to extend their capabilities beyond traditional tasks. However, learning from graphs often involves handling sensitive relationships in the data, raising significant privacy concerns. Existing privacy-preserving methods, such as DP-SGD, rely on gradient decoupling assumptions and are incompatible with relational learning due to the inherent dependencies between training samples. To address this challenge, we propose a privacy-preserving pipeline for relational learning that decouples dependencies in sampled relations for training, ensuring differential privacy through a tailored application of DP-SGD. We apply this approach to fine-tune large language models (LLMs), such as Llama2, on sensitive graph data while addressing the associated computational complexities. Our method is evaluated on four real-world text-attributed graphs, demonstrating significant improvements in relational learning tasks while maintaining robust privacy guarantees. Additionally, we analyze the trade-offs between privacy, utility, and computational efficiency, offering insights into the practical deployment of our approach for privacy-preserving relational learning. Code is available at https://github.com/Graph-COM/PvGaLM.

View on arXivView PDF

Code Implementations(1)

Graph-COM/PvGaLMOfficial100%
Apache-2.0

[COLM'25] PvGaLM is the first privacy-preserving relational learning pipeline that is compatible with DP-SGD.

32Shell, PythonOct 3, 20242 years agoApache-2.0
differential-privacyfinetuningllmsrelational-learning

Similar Papers

Cite this paper

@article{yin2024privately,
  title  = {Privately Learning from Graphs with Applications in Fine-tuning Large Language Models},
  author = {Haoteng Yin and Rongzhe Wei and Eli Chien and Pan Li},
  year   = {2024},
  eprint = {2410.08299},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.08299v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.08299.svg)](https://codeofpaper.com/paper/2410.08299)

Discussion