Upload config.json with huggingface_hub
Browse files- config.json +90 -3
config.json
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"model_type": "ResidualConvAutoencoder",
|
|
|
|
|
|
|
| 3 |
"architecture": {
|
| 4 |
"encoder_stages": 5,
|
| 5 |
"decoder_stages": 5,
|
|
@@ -15,19 +17,104 @@
|
|
| 15 |
256,
|
| 16 |
512,
|
| 17 |
512
|
| 18 |
-
]
|
|
|
|
| 19 |
},
|
| 20 |
"training": {
|
| 21 |
"dataset": "CIFAR-10",
|
|
|
|
|
|
|
| 22 |
"batch_size": 1024,
|
| 23 |
"optimizer": "AdamW",
|
| 24 |
"learning_rate": 0.001,
|
|
|
|
| 25 |
"epochs": 100,
|
| 26 |
-
"loss_function": "MSE"
|
|
|
|
| 27 |
},
|
| 28 |
"performance": {
|
| 29 |
"test_loss": 0.00429,
|
| 30 |
"validation_loss": 0.004294,
|
| 31 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
}
|
| 33 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"model_type": "ResidualConvAutoencoder",
|
| 3 |
+
"task": "image-feature-extraction",
|
| 4 |
+
"framework": "pytorch",
|
| 5 |
"architecture": {
|
| 6 |
"encoder_stages": 5,
|
| 7 |
"decoder_stages": 5,
|
|
|
|
| 17 |
256,
|
| 18 |
512,
|
| 19 |
512
|
| 20 |
+
],
|
| 21 |
+
"residual_blocks": true
|
| 22 |
},
|
| 23 |
"training": {
|
| 24 |
"dataset": "CIFAR-10",
|
| 25 |
+
"train_samples": 50000,
|
| 26 |
+
"test_samples": 10000,
|
| 27 |
"batch_size": 1024,
|
| 28 |
"optimizer": "AdamW",
|
| 29 |
"learning_rate": 0.001,
|
| 30 |
+
"weight_decay": 1e-05,
|
| 31 |
"epochs": 100,
|
| 32 |
+
"loss_function": "MSE",
|
| 33 |
+
"scheduler": "ReduceLROnPlateau"
|
| 34 |
},
|
| 35 |
"performance": {
|
| 36 |
"test_loss": 0.00429,
|
| 37 |
"validation_loss": 0.004294,
|
| 38 |
+
"train_loss": 0.003344,
|
| 39 |
+
"parameters": 34849667,
|
| 40 |
+
"training_time_minutes": 26.24
|
| 41 |
+
},
|
| 42 |
+
"preprocessing": {
|
| 43 |
+
"image_size": [
|
| 44 |
+
128,
|
| 45 |
+
128
|
| 46 |
+
],
|
| 47 |
+
"normalization": {
|
| 48 |
+
"mean": [
|
| 49 |
+
0.5,
|
| 50 |
+
0.5,
|
| 51 |
+
0.5
|
| 52 |
+
],
|
| 53 |
+
"std": [
|
| 54 |
+
0.5,
|
| 55 |
+
0.5,
|
| 56 |
+
0.5
|
| 57 |
+
]
|
| 58 |
+
},
|
| 59 |
+
"range": "[-1, 1]"
|
| 60 |
+
},
|
| 61 |
+
"calibration": {
|
| 62 |
+
"description": "Calibrated MSE reconstruction error thresholds",
|
| 63 |
+
"dataset_real": "CIFAR-10 test set",
|
| 64 |
+
"dataset_fake": "Random Noise test set",
|
| 65 |
+
"image_size": [
|
| 66 |
+
128,
|
| 67 |
+
128
|
| 68 |
+
],
|
| 69 |
+
"normalization": "mean=0.5, std=0.5 (range [-1, 1])",
|
| 70 |
+
"thresholds": {
|
| 71 |
+
"strict": {
|
| 72 |
+
"value": 0.012767740525305271,
|
| 73 |
+
"description": "High confidence - 99th percentile of real images",
|
| 74 |
+
"expected_tpr": 1.0,
|
| 75 |
+
"expected_fpr": 0.01
|
| 76 |
+
},
|
| 77 |
+
"balanced": {
|
| 78 |
+
"value": 0.009065581485629082,
|
| 79 |
+
"description": "Balanced - 95th percentile of real images",
|
| 80 |
+
"expected_tpr": 1.0,
|
| 81 |
+
"expected_fpr": 0.05
|
| 82 |
+
},
|
| 83 |
+
"sensitive": {
|
| 84 |
+
"value": 0.00931915920227766,
|
| 85 |
+
"description": "Sensitive - mean + 2*std of real images",
|
| 86 |
+
"expected_tpr": 1.0,
|
| 87 |
+
"expected_fpr": 0.0448
|
| 88 |
+
},
|
| 89 |
+
"optimal": {
|
| 90 |
+
"value": 0.20403903722763062,
|
| 91 |
+
"description": "Optimal separation between distributions",
|
| 92 |
+
"expected_tpr": 1.0,
|
| 93 |
+
"expected_fpr": 0.0
|
| 94 |
+
}
|
| 95 |
+
},
|
| 96 |
+
"statistics": {
|
| 97 |
+
"real_images": {
|
| 98 |
+
"mean": 0.004293377045542002,
|
| 99 |
+
"std": 0.0025128910783678293,
|
| 100 |
+
"median": 0.003766049398109317,
|
| 101 |
+
"min": 0.00019204482669010758,
|
| 102 |
+
"max": 0.02400028333067894,
|
| 103 |
+
"percentile_95": 0.009065581485629082,
|
| 104 |
+
"percentile_99": 0.012767740525305271
|
| 105 |
+
},
|
| 106 |
+
"fake_images": {
|
| 107 |
+
"mean": 0.40168634057044983,
|
| 108 |
+
"median": 0.4016798138618469,
|
| 109 |
+
"min": 0.3955906331539154,
|
| 110 |
+
"max": 0.4081667363643646,
|
| 111 |
+
"percentile_5": 0.3990125060081482
|
| 112 |
+
},
|
| 113 |
+
"separation": {
|
| 114 |
+
"mean_difference": 0.39739295840263367,
|
| 115 |
+
"ratio": 93.55953216552734
|
| 116 |
+
}
|
| 117 |
+
},
|
| 118 |
+
"note": "These thresholds are calibrated on CIFAR-10 vs Random Noise. For real deepfake detection, recalibrate on actual fake images."
|
| 119 |
}
|
| 120 |
}
|