Train ATI-HO and finalize project outputs
This commit is contained in:
Binary file not shown.
@@ -0,0 +1,8 @@
|
||||
method,seed,epoch,train_loss,valid_selection_loss,valid_clean_loss,lambda_interaction,lambda_mask
|
||||
A0,2026,1,1.0065954625606537,0.9363003446833118,0.931940136375008,0.001,0.0
|
||||
A0,2026,2,0.8319815562831031,0.8692282438278198,0.8585401066056975,0.001,0.0
|
||||
A0,2026,3,0.7417113295307866,0.8601733071135951,0.8481150000959963,0.001,0.0
|
||||
A0,2026,4,0.6949168112542894,0.8574190991265433,0.8474381146850166,0.001,0.0
|
||||
A0,2026,5,0.6569236450725131,0.8673936297277827,0.8559725572774698,0.001,0.0
|
||||
A0,2026,6,0.6017359914603057,0.9111922525770062,0.9027906433566586,0.001,0.0
|
||||
A0,2026,7,0.5584396239784029,0.9282659668843826,0.9144766841615949,0.001,0.0
|
||||
|
@@ -0,0 +1,137 @@
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 2026,
|
||||
"best_epoch": 4,
|
||||
"best_selection_loss": 0.8574190991265433,
|
||||
"elapsed_seconds": 6.77071833203081,
|
||||
"batch_size": 64,
|
||||
"epoch_limit": 12,
|
||||
"patience": 3,
|
||||
"optimizer": "AdamW",
|
||||
"learning_rate": 0.0003,
|
||||
"weight_decay": 0.001,
|
||||
"gradient_clip_norm": 1.0,
|
||||
"training_mask_rates": [
|
||||
0.0,
|
||||
0.1,
|
||||
0.3,
|
||||
0.5,
|
||||
0.7
|
||||
],
|
||||
"training_mask_patterns": [
|
||||
"single",
|
||||
"sync",
|
||||
"partial",
|
||||
"async"
|
||||
],
|
||||
"training_mask_seed_base": 20261227,
|
||||
"same_orders_and_masks_across_methods_for_same_seed": true,
|
||||
"config": {
|
||||
"name": "A0_main_effects",
|
||||
"low_rank": false,
|
||||
"cross_attention": false,
|
||||
"anchored": true,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0,
|
||||
"rank": 4,
|
||||
"hidden": 64,
|
||||
"gru_hidden_per_direction": 32,
|
||||
"attention_heads": 4,
|
||||
"attention_ffn": 128,
|
||||
"eta_init": 0.1
|
||||
},
|
||||
"history": [
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 2026,
|
||||
"epoch": 1,
|
||||
"train_loss": 1.0065954625606537,
|
||||
"valid_selection_loss": 0.9363003446833118,
|
||||
"valid_clean_loss": 0.931940136375008,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 2026,
|
||||
"epoch": 2,
|
||||
"train_loss": 0.8319815562831031,
|
||||
"valid_selection_loss": 0.8692282438278198,
|
||||
"valid_clean_loss": 0.8585401066056975,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 2026,
|
||||
"epoch": 3,
|
||||
"train_loss": 0.7417113295307866,
|
||||
"valid_selection_loss": 0.8601733071135951,
|
||||
"valid_clean_loss": 0.8481150000959963,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 2026,
|
||||
"epoch": 4,
|
||||
"train_loss": 0.6949168112542894,
|
||||
"valid_selection_loss": 0.8574190991265433,
|
||||
"valid_clean_loss": 0.8474381146850166,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 2026,
|
||||
"epoch": 5,
|
||||
"train_loss": 0.6569236450725131,
|
||||
"valid_selection_loss": 0.8673936297277827,
|
||||
"valid_clean_loss": 0.8559725572774698,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 2026,
|
||||
"epoch": 6,
|
||||
"train_loss": 0.6017359914603057,
|
||||
"valid_selection_loss": 0.9111922525770062,
|
||||
"valid_clean_loss": 0.9027906433566586,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 2026,
|
||||
"epoch": 7,
|
||||
"train_loss": 0.5584396239784029,
|
||||
"valid_selection_loss": 0.9282659668843826,
|
||||
"valid_clean_loss": 0.9144766841615949,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
}
|
||||
],
|
||||
"mask_counts": {
|
||||
"0.1/async": 1149,
|
||||
"0.0/async": 1181,
|
||||
"0.5/sync": 1205,
|
||||
"0.0/sync": 1177,
|
||||
"0.5/single": 1154,
|
||||
"0.0/single": 1223,
|
||||
"0.7/async": 1223,
|
||||
"0.0/partial": 1254,
|
||||
"0.7/sync": 1182,
|
||||
"0.5/partial": 1182,
|
||||
"0.3/partial": 1153,
|
||||
"0.1/partial": 1216,
|
||||
"0.3/single": 1178,
|
||||
"0.3/async": 1180,
|
||||
"0.5/async": 1182,
|
||||
"0.1/single": 1208,
|
||||
"0.7/single": 1129,
|
||||
"0.3/sync": 1220,
|
||||
"0.1/sync": 1187,
|
||||
"0.7/partial": 1182
|
||||
}
|
||||
}
|
||||
Binary file not shown.
@@ -0,0 +1,7 @@
|
||||
method,seed,epoch,train_loss,valid_selection_loss,valid_clean_loss,lambda_interaction,lambda_mask
|
||||
A0,3407,1,1.0028652648131053,0.9436100423336029,0.9391928321712619,0.001,0.0
|
||||
A0,3407,2,0.8238278020311285,0.8820641582811272,0.8730134918139532,0.001,0.0
|
||||
A0,3407,3,0.74309202587163,0.8756466648735842,0.8646609815922413,0.001,0.0
|
||||
A0,3407,4,0.6922215135009201,0.8815795684253777,0.8710692741058685,0.001,0.0
|
||||
A0,3407,5,0.6518548451088093,0.9090294297579881,0.8968785254509895,0.001,0.0
|
||||
A0,3407,6,0.6016097575150154,0.9234253629878326,0.9113659098908141,0.001,0.0
|
||||
|
@@ -0,0 +1,127 @@
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 3407,
|
||||
"best_epoch": 3,
|
||||
"best_selection_loss": 0.8756466648735842,
|
||||
"elapsed_seconds": 5.952770586998668,
|
||||
"batch_size": 64,
|
||||
"epoch_limit": 12,
|
||||
"patience": 3,
|
||||
"optimizer": "AdamW",
|
||||
"learning_rate": 0.0003,
|
||||
"weight_decay": 0.001,
|
||||
"gradient_clip_norm": 1.0,
|
||||
"training_mask_rates": [
|
||||
0.0,
|
||||
0.1,
|
||||
0.3,
|
||||
0.5,
|
||||
0.7
|
||||
],
|
||||
"training_mask_patterns": [
|
||||
"single",
|
||||
"sync",
|
||||
"partial",
|
||||
"async"
|
||||
],
|
||||
"training_mask_seed_base": 20261227,
|
||||
"same_orders_and_masks_across_methods_for_same_seed": true,
|
||||
"config": {
|
||||
"name": "A0_main_effects",
|
||||
"low_rank": false,
|
||||
"cross_attention": false,
|
||||
"anchored": true,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0,
|
||||
"rank": 4,
|
||||
"hidden": 64,
|
||||
"gru_hidden_per_direction": 32,
|
||||
"attention_heads": 4,
|
||||
"attention_ffn": 128,
|
||||
"eta_init": 0.1
|
||||
},
|
||||
"history": [
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 3407,
|
||||
"epoch": 1,
|
||||
"train_loss": 1.0028652648131053,
|
||||
"valid_selection_loss": 0.9436100423336029,
|
||||
"valid_clean_loss": 0.9391928321712619,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 3407,
|
||||
"epoch": 2,
|
||||
"train_loss": 0.8238278020311285,
|
||||
"valid_selection_loss": 0.8820641582811272,
|
||||
"valid_clean_loss": 0.8730134918139532,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 3407,
|
||||
"epoch": 3,
|
||||
"train_loss": 0.74309202587163,
|
||||
"valid_selection_loss": 0.8756466648735842,
|
||||
"valid_clean_loss": 0.8646609815922413,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 3407,
|
||||
"epoch": 4,
|
||||
"train_loss": 0.6922215135009201,
|
||||
"valid_selection_loss": 0.8815795684253777,
|
||||
"valid_clean_loss": 0.8710692741058685,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 3407,
|
||||
"epoch": 5,
|
||||
"train_loss": 0.6518548451088093,
|
||||
"valid_selection_loss": 0.9090294297579881,
|
||||
"valid_clean_loss": 0.8968785254509895,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 3407,
|
||||
"epoch": 6,
|
||||
"train_loss": 0.6016097575150154,
|
||||
"valid_selection_loss": 0.9234253629878326,
|
||||
"valid_clean_loss": 0.9113659098908141,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
}
|
||||
],
|
||||
"mask_counts": {
|
||||
"0.7/async": 1003,
|
||||
"0.0/async": 1014,
|
||||
"0.5/sync": 1033,
|
||||
"0.1/sync": 1024,
|
||||
"0.5/partial": 996,
|
||||
"0.3/partial": 1019,
|
||||
"0.5/async": 1053,
|
||||
"0.3/single": 1003,
|
||||
"0.0/sync": 1042,
|
||||
"0.7/partial": 1079,
|
||||
"0.7/sync": 986,
|
||||
"0.0/single": 1029,
|
||||
"0.1/async": 959,
|
||||
"0.3/sync": 1055,
|
||||
"0.0/partial": 1016,
|
||||
"0.1/single": 1011,
|
||||
"0.1/partial": 988,
|
||||
"0.5/single": 1076,
|
||||
"0.7/single": 997,
|
||||
"0.3/async": 987
|
||||
}
|
||||
}
|
||||
Binary file not shown.
@@ -0,0 +1,8 @@
|
||||
method,seed,epoch,train_loss,valid_selection_loss,valid_clean_loss,lambda_interaction,lambda_mask
|
||||
A0,42,1,0.9974622616061458,0.9360404843157465,0.9310548764008743,0.001,0.0
|
||||
A0,42,2,0.8197665159349088,0.8663458909307207,0.8535200449136587,0.001,0.0
|
||||
A0,42,3,0.7383816402267527,0.8669271827726575,0.8542619085573888,0.001,0.0
|
||||
A0,42,4,0.6841275570569215,0.8643234267339601,0.850496660222064,0.001,0.0
|
||||
A0,42,5,0.6515898009141287,0.8761576001460736,0.8627192581092918,0.001,0.0
|
||||
A0,42,6,0.6044590015102316,0.9111036181777388,0.8968614366028335,0.001,0.0
|
||||
A0,42,7,0.5584419562860772,0.9260857973124955,0.9113180768358838,0.001,0.0
|
||||
|
@@ -0,0 +1,137 @@
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 42,
|
||||
"best_epoch": 4,
|
||||
"best_selection_loss": 0.8643234267339601,
|
||||
"elapsed_seconds": 12.891749234986492,
|
||||
"batch_size": 64,
|
||||
"epoch_limit": 12,
|
||||
"patience": 3,
|
||||
"optimizer": "AdamW",
|
||||
"learning_rate": 0.0003,
|
||||
"weight_decay": 0.001,
|
||||
"gradient_clip_norm": 1.0,
|
||||
"training_mask_rates": [
|
||||
0.0,
|
||||
0.1,
|
||||
0.3,
|
||||
0.5,
|
||||
0.7
|
||||
],
|
||||
"training_mask_patterns": [
|
||||
"single",
|
||||
"sync",
|
||||
"partial",
|
||||
"async"
|
||||
],
|
||||
"training_mask_seed_base": 20261227,
|
||||
"same_orders_and_masks_across_methods_for_same_seed": true,
|
||||
"config": {
|
||||
"name": "A0_main_effects",
|
||||
"low_rank": false,
|
||||
"cross_attention": false,
|
||||
"anchored": true,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0,
|
||||
"rank": 4,
|
||||
"hidden": 64,
|
||||
"gru_hidden_per_direction": 32,
|
||||
"attention_heads": 4,
|
||||
"attention_ffn": 128,
|
||||
"eta_init": 0.1
|
||||
},
|
||||
"history": [
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 42,
|
||||
"epoch": 1,
|
||||
"train_loss": 0.9974622616061458,
|
||||
"valid_selection_loss": 0.9360404843157465,
|
||||
"valid_clean_loss": 0.9310548764008743,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 42,
|
||||
"epoch": 2,
|
||||
"train_loss": 0.8197665159349088,
|
||||
"valid_selection_loss": 0.8663458909307207,
|
||||
"valid_clean_loss": 0.8535200449136587,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 42,
|
||||
"epoch": 3,
|
||||
"train_loss": 0.7383816402267527,
|
||||
"valid_selection_loss": 0.8669271827726575,
|
||||
"valid_clean_loss": 0.8542619085573888,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 42,
|
||||
"epoch": 4,
|
||||
"train_loss": 0.6841275570569215,
|
||||
"valid_selection_loss": 0.8643234267339601,
|
||||
"valid_clean_loss": 0.850496660222064,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 42,
|
||||
"epoch": 5,
|
||||
"train_loss": 0.6515898009141287,
|
||||
"valid_selection_loss": 0.8761576001460736,
|
||||
"valid_clean_loss": 0.8627192581092918,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 42,
|
||||
"epoch": 6,
|
||||
"train_loss": 0.6044590015102316,
|
||||
"valid_selection_loss": 0.9111036181777388,
|
||||
"valid_clean_loss": 0.8968614366028335,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
},
|
||||
{
|
||||
"method": "A0",
|
||||
"seed": 42,
|
||||
"epoch": 7,
|
||||
"train_loss": 0.5584419562860772,
|
||||
"valid_selection_loss": 0.9260857973124955,
|
||||
"valid_clean_loss": 0.9113180768358838,
|
||||
"lambda_interaction": 0.001,
|
||||
"lambda_mask": 0.0
|
||||
}
|
||||
],
|
||||
"mask_counts": {
|
||||
"0.0/single": 1169,
|
||||
"0.3/single": 1187,
|
||||
"0.5/single": 1158,
|
||||
"0.7/sync": 1272,
|
||||
"0.7/partial": 1207,
|
||||
"0.0/async": 1145,
|
||||
"0.3/async": 1200,
|
||||
"0.3/partial": 1209,
|
||||
"0.3/sync": 1170,
|
||||
"0.1/single": 1205,
|
||||
"0.5/async": 1185,
|
||||
"0.7/single": 1232,
|
||||
"0.5/sync": 1143,
|
||||
"0.5/partial": 1236,
|
||||
"0.0/partial": 1183,
|
||||
"0.0/sync": 1159,
|
||||
"0.7/async": 1179,
|
||||
"0.1/sync": 1202,
|
||||
"0.1/async": 1192,
|
||||
"0.1/partial": 1132
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user