=== Summary ===
{
  "baseline_full": {
    "acc": 0.51875,
    "auc": 0.5556660666066607,
    "avg_kept": 64.0
  },
  "baseline_attn_prune": {
    "acc": 0.5225,
    "auc": 0.5559348434843485,
    "avg_kept": 32.0
  },
  "proposed": {
    "acc": 0.55125,
    "auc": 0.5560973597359735,
    "avg_kept": 32.0
  }
}

Efficiency:
{
  "flops_full": 1048576.0,
  "flops_proposed": 262144.0,
  "latency_full_ms": 83.92,
  "latency_proposed_ms": 22.48,
  "keep_rate": 0.5
}