Skip to content

Judge tuning (2/5): Add held-out splits and NePS configuration - #135

Open
ErlisLushtaku wants to merge 19 commits into
erlislushtaku/fix/vllm-request-contextfrom
erlislushtaku/feat/judge-tuning-02-neps-config
Open

ErlisLushtaku wants to merge 19 commits into
erlislushtaku/fix/vllm-request-contextfrom
erlislushtaku/feat/judge-tuning-02-neps-config

Conversation

@ErlisLushtaku

@ErlisLushtaku ErlisLushtaku commented Oct 7, 2026 •

Copy link
Copy Markdown
Collaborator

Adds the held-out split and the NePS configuration that the tuning runner in #133 builds on.

  • meta_eval.split (validation or test, default all) restricts sampling to one half of the arena prompts, meta_eval.validation_fraction sets the share (default 0.5). The split hashes the prompt text, so duplicate prompts stay on one side and the partition does not depend on run.seed. Tuning selects on validation and reports on test, so a different seed must not move battles across.
  • A packaged tune-judge task with defaults: meta-eval-lmarena-140k-en as the target, agreement and cost objectives, mo_hyperband and 10 to 90 battles per model. Trials run under the target meta-eval task, so they reuse its defaults and judgement cache.
  • tune_judge.search_space is nested like the run config. A list is categorical and {lower, upper} is a range whose type follows the config field. The reference judge is the prior unless prior is set.
  • tune_judge.run_dir selects a directory to resume or share between workers, and a config hash rejects resuming with different search settings. NePS state stays under neps/, apart from the inference cache.
  • The tune extra pins NePS 0.17.0. Download coverage skips tune-judge, which has no dataset. CI also runs for PRs based on another stack branch.

Merge order: #138 → #135 → #140 → #133 → #136.

Judge tuning needs to select configurations on one partition of the
human-labeled arena and report on another. Battles are assigned by a
salted hash of their first user prompt, so repeated prompts never cross
partitions and changing run.seed cannot move battles between them.
@ErlisLushtaku ErlisLushtaku changed the title WIP: Judge tuning (2/4): Add NePS configuration and sessions WIP: Judge tuning (2/5): Add NePS configuration and sessions Oct 9, 2026
Comment thread judgearena/tasks/schema/task.py Outdated
@ErlisLushtaku ErlisLushtaku changed the title WIP: Judge tuning (2/5): Add NePS configuration and sessions WIP: Judge tuning (2/6): Add NePS configuration and sessions Oct 9, 2026
@ErlisLushtaku ErlisLushtaku changed the title WIP: Judge tuning (2/6): Add NePS configuration and sessions WIP: Judge tuning (2/5): Add held-out splits and NePS configuration Oct 13, 2026
@ErlisLushtaku
ErlisLushtaku changed the base branch from erlislushtaku/feat/judge-tuning-01-arena-split to erlislushtaku/fix/vllm-request-context October 13, 2026 12:07
@ErlisLushtaku ErlisLushtaku changed the title WIP: Judge tuning (2/5): Add held-out splits and NePS configuration Judge tuning (2/5): Add held-out splits and NePS configuration Oct 13, 2026
@ErlisLushtaku
ErlisLushtaku marked this pull request as ready for review October 13, 2026 12:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant