The Llama model attention map with 3 documents is represented as follows: ./visualization-tools/vis.ipynb reproduces the visualization results in the paper. We provide more visualization tools under .

Abstract: Federated learning (FL) enables collaborative training of a machine learning (ML) model across multiple parties, facilitating the preservation of users’ and institutions’ privacy by ...