config.json

{
  "model_type": "BADClassifier",
  "input_dim": 2048,
  "output_dim": 2,
  "hidden_size": 2048,
  "target_layer": 15,
  "num_layers": 22,
  "base_model_name": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
  "label_mapping": {
    "biased": 0,
    "unbiased": 1
  },
  "dataset_info": {
    "bbq_samples": 58942,
    "mmlu_samples": 20266,
    "bbq_source": "bitlabsdb/BBQ_dataset",
    "mmlu_source": "cais/mmlu"
  },
  "layer_selection": {
    "candidates": [
      11,
      12,
      13,
      14,
      15,
      16
    ],
    "selected": 15,
    "accuracies": {
      "11": 0.8152321577072144,
      "12": 0.8395493626594543,
      "13": 0.8271411657333374,
      "14": 0.8292092084884644,
      "15": 0.8414747714996338,
      "16": 0.8393354415893555
    }
  },
  "performance": {
    "best_epoch": 50
  },
  "trained_date": "2025-11-04 13:03:45"
}