Repository navigation
Add ConfAIde contextual privacy benchmark - #4410
Open
barkha1research-ai wants to merge 1 commit into
Open
barkha1research-ai wants to merge 1 commit into
barkha1research-ai wants to merge 1 commit into
Conversation
Signed-off-by: Barkha <340367068+barkha1research-ai@users.noreply.github.com>
This branch has not been deployed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Closes #4405.
Adds ConfAIde (Mireshghallah et al., ICLR 2024), a benchmark for whether models can tell when it is appropriate to share information, based on contextual integrity. As far as I can tell this is the first privacy benchmark in the harness.
What's added
lm_eval/tasks/confaide/: nine tasks and aconfaidegroup, one task per tier and question type.confaide_tier1,confaide_tier2a,confaide_tier2b: Pearson correlation of the model's ratings with the human ratings.confaide_tier3_free_response,confaide_tier3_info_accessibility,confaide_tier3_privacy_sharing,confaide_tier3_control: the reference's string-match leakage and error rates.confaide_tier4_meeting_summary,confaide_tier4_action_item: error when the public item is missing or the private one appears.eval.py; the README lists where they differ (no Llama-2 judge metric,max_gen_toks, no re-querying of unparseable ratings)._worst_caseform (an item fails if any sample fails), which the paper uses for Tier 3.tests/test_confaide.py: parsing, prompt construction and scoring, using small fixtures in the benchmark's own file formats (no downloads).Validation
I ran the reference
eval.py(with its model backends stubbed out) and this task on the same responses: a deterministic fake model returned the same canned responses throughsimple_evaluate, and the reference scored them directly. For all 1,326 documents, with and without the privacy prompt, the prompts, gold fields and per-document metrics were identical, and the aggregates agreed to within floating-point summation order. The tests pass on Python 3.10, 3.12 and 3.14.Results for three small open models, run through Ollama with
local-chat-completionsand--apply_chat_template, one sample per prompt:These are single-sample numbers, because the API backends send the same seed with every request, so Ollama returns the same response for all 10 repeats (noted in the README). Llama 3.1 8B often refuses ("I cannot provide…"): 57 of its 98 Tier 2a ratings could not be parsed and count as 0, as in the reference, and most of its Tier 3 control errors are refusals.
Notes
new_tasksCI job may reportDuplicate task 'confaide_…' found in both 'confaide_…' and 'confaide'when a group file and its tasks are both in the changed-files list. The same happens locally with existing groups such askormedmcqa, so it isn't specific to this change. With only the nine task files in that list,tests/test_tasks.pypasses locally (120 tests).pre-commitpasses on the changed files.