outputs/student/selection.json

{
  "keep_idx": [
    12,
    13,
    14,
    15,
    16,
    17,
    18,
    19
  ],
  "surrogate_idx": [
    0,
    1,
    2,
    3,
    4,
    5,
    6,
    7,
    8,
    9,
    10,
    11
  ],
  "rank": 512,
  "act": "gelu",
  "lstsq_rel_err": {
    "0": 0.923846423625946,
    "1": 0.9222169518470764,
    "2": 0.9311394691467285,
    "3": 0.9383108615875244,
    "4": 0.9520511031150818,
    "5": 0.9648131132125854,
    "6": 0.952173113822937,
    "7": 0.9317150712013245,
    "8": 0.9289534687995911,
    "9": 0.9211865663528442,
    "10": 0.9375649094581604,
    "11": 0.9256151914596558
  },
  "stats": [
    {
      "block": 0,
      "tail_energy": 59089140.474642806,
      "captured_ratio": 0.9668478738420475,
      "delta_rms": 6.9533952156469985
    },
    {
      "block": 1,
      "tail_energy": 83874863.38740343,
      "captured_ratio": 0.9417839860523992,
      "delta_rms": 6.251632657927463
    },
    {
      "block": 2,
      "tail_energy": 114292191.07172194,
      "captured_ratio": 0.9416515796108982,
      "delta_rms": 7.289412025781583
    },
    {
      "block": 3,
      "tail_energy": 119547086.05953766,
      "captured_ratio": 0.9502712670401754,
      "delta_rms": 8.075409820859075
    },
    {
      "block": 4,
      "tail_energy": 116658598.19075358,
      "captured_ratio": 0.9655388243757796,
      "delta_rms": 9.582792923706565
    },
    {
      "block": 5,
      "tail_energy": 88129576.21833016,
      "captured_ratio": 0.9829519899930255,
      "delta_rms": 11.841937547872696
    },
    {
      "block": 6,
      "tail_energy": 233947614.2785113,
      "captured_ratio": 0.9480229411771703,
      "delta_rms": 11.049744377345622
    },
    {
      "block": 7,
      "tail_energy": 239045574.6241713,
      "captured_ratio": 0.9230449688797725,
      "delta_rms": 9.17953360989325
    },
    {
      "block": 8,
      "tail_energy": 213803023.09700704,
      "captured_ratio": 0.8946195050314197,
      "delta_rms": 7.4186618350838875
    },
    {
      "block": 9,
      "tail_energy": 235987032.89428294,
      "captured_ratio": 0.8796560983338009,
      "delta_rms": 7.293412842043697
    },
    {
      "block": 10,
      "tail_energy": 196456732.34142846,
      "captured_ratio": 0.9061375520473004,
      "delta_rms": 7.535052648165694
    },
    {
      "block": 11,
      "tail_energy": 351052551.9025931,
      "captured_ratio": 0.8558331244669359,
      "delta_rms": 8.127408622254622
    },
    {
      "block": 12,
      "tail_energy": 369463408.51814806,
      "captured_ratio": 0.8495481370185651,
      "delta_rms": 8.161795381133864
    },
    {
      "block": 13,
      "tail_energy": 364266719.28867227,
      "captured_ratio": 0.8597546764198015,
      "delta_rms": 8.393910303472095
    },
    {
      "block": 14,
      "tail_energy": 503035651.9435967,
      "captured_ratio": 0.8524276232651644,
      "delta_rms": 9.616025767874618
    },
    {
      "block": 15,
      "tail_energy": 592747358.497119,
      "captured_ratio": 0.8497173970186442,
      "delta_rms": 10.34377894136027
    },
    {
      "block": 16,
      "tail_energy": 711066588.7863977,
      "captured_ratio": 0.8706560150807409,
      "delta_rms": 12.211831106208086
    },
    {
      "block": 17,
      "tail_energy": 1067588457.5499623,
      "captured_ratio": 0.881381903540417,
      "delta_rms": 15.62518726400348
    },
    {
      "block": 18,
      "tail_energy": 1865590107.5882843,
      "captured_ratio": 0.9082888026675959,
      "delta_rms": 23.49068627097329
    },
    {
      "block": 19,
      "tail_energy": 359424342.8436452,
      "captured_ratio": 0.9961650619514839,
      "delta_rms": 50.42237972602075
    }
  ],
  "param_summary": {
    "total_params": 2441089536,
    "total_B": 2.441089536,
    "n_double": 5,
    "n_full_single": 8,
    "n_surrogate": 12
  }
}