Summary
This paper proposes a novel probing tool to probe potential privacy leakage in large language models. The proposed method includes both black-box and white-box probing. The authors demonstrate its effectiveness on the OPT-1.3B model with the publicly available Pile dataset. The experimental results show that the proposed method can probe the leakage of personally identifiable information (PII) in the Pile dataset, while the white-box probing generally achieves better performance. The results prove that PII leakage could be a serous threat to LLM’s security, calling for more research attention in this area.
Strengths
1. Timely and important topic. With the more and often usage of LLMs, LLMs’ security and potential PII leakage should be taken into account. This paper proposes an early attempt in this area regarding probing PII from LLMs, showing the importance of the research problem.
2. Novel method. The authors propose a novel probing tool including both black-box and white-box probing.
3. Well-written paper. The paper is generally well-written with good presentation quality. I enjoy reading the paper.
Weaknesses
I generally do not find significant weaknesses in this paper. I do have a few small concerns, that, I wish the authors could address in the revision.
1. The definition of the soft prompt \theta_s is not clear (Line 172).
2. Evaluation. In the current form, only one model was included for evaluation. IMO this is fine for demonstration. But if the authors would like to improve it, more models could be considered.
A follow-up comment regarding the candidate LLMs: The selection criteria (Line 205-211) should be more clear, and the selection should have its rationale, e.g., is OPT-1.3B the only model that is available for evaluation? If there are more candidates, why OPT-1.3B is the only one chosen? All these details can help improve the paper’s soundness and significance.
3. Lack of examples. It would be nice to add a few examples in the paper to better show the probing results (both black-box and white-box ones).
Questions
1. Is it possible to extend the proposed method to more LLMs or even closed-source LLMs such as ChatGPT or GPT4?
2. How do you justify the choice of OPT-1.3B as the subject LLM for evaluation?
Rating
7: Accept: Technically solid paper, with high impact on at least one sub-area, or moderate-to-high impact on more than one areas, with good-to-excellent evaluation, resources, reproducibility, and no unaddressed ethical considerations.
Confidence
4: You are confident in your assessment, but not absolutely certain. It is unlikely, but not impossible, that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work.
Limitations
There are discussions about the limitations and potential social impacts of this paper. I have two more comments.
1. Regarding the limitation, it would be nice to mention the potential generalization ability of the proposed method to other LLMs (or even closed-source LLMs, such as GPT4)
2. Regarding the potential social impact, the authors should directly point out that the research outcome (i.e., the probing tool) might potentially be used for undesired purposes and potentially lead to negative social impact.