Perturbation-Based Data Selection for Medical Question Answering
Organizations: Department of Computer Science, University of Memphis, Memphis, TN, USA
Abstract
Selecting a compact training subset is an important step in adapting medical language models from existing question-answering banks. We investigate perturbation-based data selection using two signals derived from Gaussian perturbations of the final-block representation: prediction-change frequency and non-flipping margin contraction. After a shared entropy prefilter, Representation-Sensitivity Prioritization (RSP) combines the signals using automatic root-mean-square scaling, while RSP-Drop ranks candidates by contraction alone. Both selectors use only questions and answer options during selection. We evaluate them with MedGemma-4B on five medical QA datasets at retained-set budgets of 64, 128, and 256 examples. At the primary budget of 128, RSP and RSP-Drop obtain dataset-averaged Macro-F1 scores of 47.55 and 48.28, respectively, compared with 45.11 for entropy-pool-matched random selection. Their relative performance varies across datasets and budgets. A selected-set case study identifies contrasting clean-margin profiles among the examples exchanged by the two rules, while a full-pool ablation shows that entropy prefiltering is not uniformly beneficial. These retrospective results characterize the empirical utility and limitations of perturbation-based selection under a fixed adaptation protocol. Evaluation uses one backbone and benchmarks with documented training-family overlap; it does not establish clinical effectiveness.
Figures & tables
| Dataset | Answer format | Pool | Dev | Test |
|---|---|---|---|---|
| AfriMed-QA | 5 options | 1,500 | 559 | 892 |
| MedExpQA | 4/5 options | 434 | 63 | 121 |
| MedExQA | 4 options | 600 | 164 | 200 |
| PubMedQA | Yes/no/maybe | 600 | 200 | 200 |
| MedMCQA | 4 options | 512 | 256 | 4,096 |
| Method | Training seconds | Input tokens | Optimizer updates | |
|---|---|---|---|---|
| 64 | RSP | 43.55 | 32.47 | 16 |
| 64 | Random@ | 43.48 | 34.45 | 16 |
| 128 | RSP | 86.72 | 64.42 | NA |
| 128 | Random@ | 87.30 | 69.41 | NA |
| 256 | RSP | 172.86 | 130.96 | 64 |
| 256 | Random@ | 172.78 | 137.58 | 64 |
| Method | AfriMed-QA | MedExpQA | MedExQA | PubMedQA | MedMCQA | Dataset-macro avg. |
|---|---|---|---|---|---|---|
| Question/options-only controls | ||||||
| Random@ | 49.36 / 46.04 | 51.79 / 49.64 | 50.33 / 49.64 | 69.50 / 48.58 | 36.09 / 31.68 | 51.42 / 45.11 |
| Entropy | 50.34 / 47.19 | 51.24 / 49.00 | 48.67 / 48.60 | 70.00 / 50.74 | 38.00 / 33.52 | 51.65 / 45.81 |
| Margin | 51.49 / 49.34 | 52.07 / 48.63 | 48.67 / 48.54 | 69.00 / 49.17 | 38.48 / 34.72 | 51.94 / 46.08 |
| BALD | 53.33 / 50.86 | 50.96 / 46.99 | 48.67 / 47.83 | 69.17 / 48.79 | 37.64 / 35.64 | 51.95 / 46.02 |
| BatchBALD | 49.96 / 46.66 | 50.14 / 45.88 | 49.50 / 49.19 | 71.67 / 54.34 | 37.65 / 35.00 | 51.78 / 46.21 |
| Method | Seed 13 | Seed 23 | Seed 37 |
|---|---|---|---|
| Random@ | 51.60 / 45.56 | 50.23 / 43.89 | 52.42 / 45.90 |
| Entropy | 51.24 / 45.46 | 53.29 / 47.93 | 50.41 / 44.04 |
| Margin | 52.35 / 46.78 | 50.65 / 45.16 | 52.83 / 46.31 |
| BALD | 52.36 / 45.55 | 51.36 / 45.62 | 52.13 / 46.90 |
| BatchBALD | 53.90 / 50.06 | 50.86 / 43.79 | 50.59 / 44.79 |
| Noise-Stability | 49.48 / 44.16 | 51.32 / 45.17 | 51.63 / 45.41 |
| Exclusive group | Mean clean margin | |||
|---|---|---|---|---|
| RSP only | 26 | 1 | 0 | 0.098 |
| RSP-Drop only | 26 | 0 | 0.707 |
| Comparison | Accuracy | Macro-F1 | |
|---|---|---|---|
| 64 | RSP – Random@ | ||
| 64 | RSP-Drop – Random@ | ||
| 64 | RSP-Drop – RSP | ||
| 128 | RSP – Random@ | ||
| 128 | RSP-Drop – Random@ | ||
| 128 | RSP-Drop – RSP |
| Method | AfriMed-QA | MedExpQA | MedExQA | PubMedQA | MedMCQA | Dataset-macro avg. | |
|---|---|---|---|---|---|---|---|
| 64 | RSP | 48.43 / 43.52 | 45.73 / 41.35 | 44.50 / 44.37 | 68.17 / 46.75 | 35.42 / 31.06 | 48.45 / 41.41 |
| 64 | RSP-Drop | 47.83 / 44.13 | 48.21 / 45.29 | 44.17 / 44.16 | 65.17 / 44.39 | 36.39 / 32.19 | 48.35 / 42.03 |
| 64 | Random@ | 48.13 / 44.12 | 45.45 / 42.34 | 43.00 / 42.70 | 64.00 / 43.18 | 36.07 / 31.73 | 47.33 / 40.81 |
| 128 | RSP | 53.70 / 51.26 | 51.24 / 49.48 | 52.50 / 52.31 | 70.00 / 51.00 | 37.54 / 33.72 | 53.00 / 47.55 |
| 128 | RSP-Drop | 52.39 / 50.11 | 53.72 / 50.82 | 54.50 / 54.28 | 68.83 / 49.92 | 38.56 / 36.28 | 53.60 / 48.28 |
| 128 | Random@ | 49.36 / 46.04 | 51.79 / 49.64 | 50.33 / 49.64 | 69.50 / 48.58 | 36.09 / 31.68 | 51.42 / 45.11 |
| Dataset | Method | Top- / full pool | Entropy-prefiltered | Full-pool | ||
|---|---|---|---|---|---|---|
| 64 | AfriMed-QA | RSP | 256 / 1,500 | 48.43 / 43.52 | 46.82 / 42.09 | +1.61 / +1.44 |
| RSP-Drop | 47.83 / 44.13 | 46.79 / 42.53 | +1.05 / +1.59 | |||
| 64 | MedExpQA | RSP | 256 / 434 | 45.73 / 41.35 | 46.01 / 42.33 | -0.28 / -0.98 |
| RSP-Drop | 48.21 / 45.29 | 46.28 / 42.32 | +1.93 / +2.97 | |||
| 64 | MedExQA | RSP | 256 / 600 | 44.50 / 44.37 | 42.33 / 42.61 | +2.17 / +1.76 |
| RSP-Drop | 44.17 / 44.16 | 44.33 / 43.51 | -0.17 / +0.65 |
Data Selection
Data selection chooses informative, representative, or reliable examples to improve model training and evaluation while reducing data, compute, or labeling costs.
See all 82 papers