Skip to content

Add KMMLU (Korean-native MMLU, 45 subjects) - #1425

Open
CY-HYUN wants to merge 1 commit into
huggingface:mainfrom
CY-HYUN:add-kmmlu
Open

CY-HYUN wants to merge 1 commit into
huggingface:mainfrom
CY-HYUN:add-kmmlu

Conversation

@CY-HYUN

@CY-HYUN CY-HYUN commented Oct 10, 2026

Copy link
Copy Markdown

Adds KMMLU (HAERAE-HUB/KMMLU), a Korean multiple-choice benchmark built from original Korean exams: 35,030 questions in 45 subjects (arXiv 2402.11548). This follows the proposal in #1424.

What it adds

One file, src/lighteval/tasks/tasks/kmmlu.py: 45 tasks named kmmlu:<subject> (for example kmmlu:accounting), one per HF config, test split, loglikelihood_acc.

The prompt copies the default kmmlu task of lm-evaluation-harness (v0.4.13), so scores from the two tools can be compared: the question, the four options as A. to D., then 정답: (full-width colon), choices " A" to " D", gold answer - 1. The dataset revision is pinned (d61b3f1). In #1424 I asked whether you'd rather have the multilingual template style (get_mcq_prompt_function, like cmmlu). Happy to switch if you prefer it.

How I checked it

I ran this task as a --custom-tasks file with lighteval 0.13.0, ran lm-eval 0.4.13 on the same model, and compared each question. Setup: Qwen/Qwen2.5-0.5B-Instruct (revision 7ae5576), float32, CPU, batch size 1, 0-shot, chat template off.

Subject n lighteval acc lm-eval acc prompts differ predictions differ max abs log-prob diff
Health 100 0.3300 0.3300 0 0 5.15e-05
Education 100 0.3600 0.3600 0 0 5.53e-05
Korean-History 100 0.2300 0.2300 0 0 4.01e-05
Accounting 100 0.2500 0.2500 0 0 5.05e-05
Math 300 0.2733 0.2733 0 0 4.24e-05

All 700 questions got the same prompt, tokens and predicted choice. Code and a reproduce.py that prints this table: https://github.com/CY-HYUN/kmmlu-lighteval

On main (7fa19dc) I checked that the config fields and Metrics.loglikelihood_acc the file uses are unchanged, and ran ruff check and ruff format with the repo's config (ruff 0.11.10). I haven't run it from a source install of main.

Limits

  • 5 of 45 subjects, one small model, CPU only. A full 45-subject run on GPU isn't done yet.
  • Only the log-likelihood variant. lm-eval also has kmmlu_direct, kmmlu_hard and chain-of-thought variants.
  • lighteval tokenizes prompt and choice separately and lm-eval tokenizes them together, so they agree only when no token crosses the boundary. That held for the Qwen2.5 tokenizer on these 700 questions.

I used an AI coding assistant (Claude Code) to write and check this.

Copilot AI balanced review requested due to automatic review settings October 10, 2026 19:39

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copilot was unable to review this pull request because the user who requested the review has reached their quota limit.

CY-HYUN added a commit to CY-HYUN/kmmlu-lighteval that referenced this pull request Oct 10, 2026
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.

2 participants