aiBatteryLifeCycle / v3 /models.json
NeerajCodz's picture
fix: v3
c8de822
{
"version": "v3",
"display": "v3.0",
"description": "Production-grade models with cross-battery generalization split, 18 physics-informed features, and optimized hyperparameters. Highest accuracy across all versions.",
"split_strategy": "cross-battery grouped split (no data leakage)",
"features": 18,
"feature_set": [
"cycle_number",
"ambient_temperature",
"peak_voltage",
"min_voltage",
"voltage_range",
"avg_current",
"avg_temp",
"temp_rise",
"cycle_duration",
"Re",
"Rct",
"delta_capacity",
"capacity_retention",
"cumulative_energy",
"dRe_dn",
"dRct_dn",
"soh_rolling_mean",
"voltage_slope"
],
"sequence_length": 32,
"dataset": "NASA PCoE Li-ion 18650 (30 batteries, 2678 cycles)",
"default_model": "best_ensemble",
"models": {
"xgboost": {
"display_name": "XGBoost",
"family": "classical",
"algorithm": "XGBRegressor",
"version": "3.0",
"requires_scaling": false,
"file": "models/classical/xgboost.joblib",
"sha256": "879f93f0d4dcc126b996c76466907add4edcea0c3c5506566fb364d1fa1ea858",
"bytes": 10991665,
"r2": 0.9456846914579201,
"mae": 1.4743979527641207,
"f1_macro": 0.6718181818181819,
"f1_weighted": 0.958783699059561
},
"gradient_boosting": {
"display_name": "GradientBoosting",
"family": "classical",
"algorithm": "GradientBoostingRegressor",
"version": "3.0",
"requires_scaling": false,
"file": "models/classical/gradient_boosting.joblib",
"sha256": "17ae16a3244e10d860ed152d16eaa491d75fce5543b3ac9cf60e2a689a45f404",
"bytes": 5380250,
"r2": 0.9859839154554527,
"mae": 1.383230362734403,
"within_5pct": 95.14563106796116,
"f1_macro": 0.8881706137824349,
"f1_weighted": 0.9370460410953712
},
"lightgbm": {
"display_name": "LightGBM",
"family": "classical",
"algorithm": "LGBMRegressor",
"version": "3.0",
"requires_scaling": false,
"file": "models/classical/lightgbm.joblib",
"sha256": "2654a3fa80438261bda6b49cb47f7c41c80756810ae6bad3c0e91617ff286fc7",
"bytes": 9270258,
"r2": 0.9825540467137408,
"mae": 1.9767818835386108,
"within_5pct": 89.51456310679612,
"f1_macro": 0.8704860258391961,
"f1_weighted": 0.9279456806427443
},
"random_forest": {
"display_name": "Random Forest",
"family": "classical",
"algorithm": "RandomForestRegressor",
"version": "3.0",
"requires_scaling": false,
"file": "models/classical/random_forest.joblib",
"sha256": "dd1535f732486f70ddc881f1a5d48d2ed43735e2bf932cd3d985a1c72b73cc0d",
"bytes": 86125777,
"r2": 0.9452882771430996,
"mae": 1.4566859966683918,
"f1_macro": 0.6851386011640525,
"f1_weighted": 0.9643882261296104
},
"extra_trees": {
"display_name": "ExtraTrees",
"family": "classical",
"algorithm": "ExtraTreesRegressor",
"version": "3.0",
"requires_scaling": false,
"file": "models/classical/extra_trees.joblib",
"sha256": "9a69db604e6a7e008a063309b3760ab7b949f234a220821bb12e6cf56b37da77",
"bytes": 133683937,
"r2": 0.9179773492318436,
"mae": 3.0058840278633383,
"f1_macro": 0.4795673076923077,
"f1_weighted": 0.7813163129973475
},
"svr": {
"display_name": "SVR (RBF)",
"family": "classical",
"algorithm": "SVR",
"version": "3.0",
"requires_scaling": true,
"file": "models/classical/svr.joblib",
"sha256": "8b94d7f86b8adb067870435097bb87e541aeda9baefc35bb8332f1c29d3cc709",
"bytes": 225246,
"r2": 0.889796092634333,
"mae": 4.919579919363492,
"within_5pct": 79.2233009708738,
"f1_macro": 0.7124677351662732,
"f1_weighted": 0.7837896622370352
},
"ridge": {
"display_name": "Ridge Regression",
"family": "classical",
"algorithm": "Ridge",
"version": "3.0",
"requires_scaling": true,
"file": "models/classical/ridge.joblib",
"sha256": "6241bd2746d8fdbb611401f079949491fdb1cfef35c9fa29bae021cecb71090b",
"bytes": 697,
"r2": 0.9647019468733783,
"mae": 3.2588075991875645,
"within_5pct": 88.93203883495146,
"f1_macro": 0.7797303637024042,
"f1_weighted": 0.8371654548106269
},
"knn_k5": {
"display_name": "KNN (k=5)",
"family": "classical",
"algorithm": "KNeighborsRegressor",
"version": "3.0",
"requires_scaling": true,
"file": "models/classical/knn_k5.joblib",
"sha256": "7786c46318c8c5d9c41ad5ab37722289088cacbfda57534cf8450c000a588e8d",
"bytes": 329444,
"r2": 0.7554756946432488,
"mae": 11.023402656799151,
"within_5pct": 34.1747572815534,
"f1_macro": 0.4414071556445377,
"f1_weighted": 0.4562966979325331
},
"vanilla_lstm": {
"display_name": "Vanilla LSTM",
"family": "deep_pytorch",
"algorithm": "VanillaLSTM",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/vanilla_lstm.pt",
"sha256": "ffb4717050bebb7c3f2a16528dcf8f909d02f4bc33726abbae6c842ccf66bc4f",
"bytes": 834487,
"r2": 0.7143583597773893,
"mae": 8.20209840058283
},
"bidirectional_lstm": {
"display_name": "Bidirectional LSTM",
"family": "deep_pytorch",
"algorithm": "BidirectionalLSTM",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/bidirectional_lstm.pt",
"sha256": "60aabaf65ebaa709e97e29d7c72e189b7f2157d1a4607d05d461b053cebe5352",
"bytes": 2191585,
"r2": 0.783251920455803,
"mae": 6.736083068479388
},
"gru": {
"display_name": "GRU",
"family": "deep_pytorch",
"algorithm": "GRUModel",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/gru.pt",
"sha256": "38b5af307d2d53cd6c9ca385b8a3f9350d558613fb9f236429902f00a67e012e",
"bytes": 626424,
"r2": 0.7281985216161616,
"mae": 7.84354667939861
},
"attention_lstm": {
"display_name": "Attention LSTM",
"family": "deep_pytorch",
"algorithm": "AttentionLSTM",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/attention_lstm.pt",
"sha256": "d08b1e02e0328925f0c105398226bacbf1efcf2ccf850a23dab6fd176fc3a9cc",
"bytes": 1463313,
"r2": 0.24013211796996825,
"mae": 14.60765696652663
},
"batterygpt": {
"display_name": "BatteryGPT",
"family": "deep_pytorch",
"algorithm": "BatteryGPT",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/batterygpt.pt",
"sha256": "f4bcca951f4d32c48a1f731359f999e83f22ae3ac95480d7ef2a084844122587",
"bytes": 432034,
"r2": 0.6258803704512991,
"mae": 9.200522572264914
},
"tft": {
"display_name": "Temporal Fusion Transformer",
"family": "deep_pytorch",
"algorithm": "TemporalFusionTransformer",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/tft.pt",
"sha256": "0c9aa7246ce7974c11d1e441e28b954df0ffd5a189f3692a14ccd24559294abc",
"bytes": 7265508,
"r2": 0.8983696122990692,
"mae": 2.3153114237171377,
"f1_macro": 0.6755038661975573,
"f1_weighted": 0.9681951413747805
},
"itransformer": {
"display_name": "iTransformer",
"family": "deep_keras",
"algorithm": "iTransformer",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/itransformer.keras",
"sha256": "4812ea0ce7757d9f737c8487162f0097d9574ac0d22cc6d032da174fc14d5d8d",
"bytes": 1708244,
"r2": 0.32596745079552714,
"mae": 12.580708086984522
},
"physics_itransformer": {
"display_name": "Physics iTransformer",
"family": "deep_keras",
"algorithm": "PhysicsITransformer",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/physics_itransformer.keras",
"sha256": "ea915e5091e1f8e7150adddc76ec036c3ff73d56fd1250fb422f42a34c33cba0",
"bytes": 1755582,
"r2": 0.46946799132396844,
"mae": 10.154041720377204
},
"dynamic_graph_itransformer": {
"display_name": "DG-iTransformer",
"family": "deep_keras",
"algorithm": "DynamicGraphITransformer",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/dynamic_graph_itransformer.keras",
"sha256": "a38fa4831fb1c4c840b04d8b1e2372bb54cb9640696ef207dfc59a1165207272",
"bytes": 1723503,
"r2": 0.7596155006117086,
"mae": 6.491724032102822
},
"vae_lstm": {
"display_name": "VAE-LSTM",
"family": "deep_pytorch",
"algorithm": "VAE_LSTM",
"version": "3.0",
"requires_scaling": true,
"file": "models/deep/vae_lstm.pt",
"sha256": "a0f0023c697c94bc49e11666e0e0cdcdd92256b6c6d234043834ce27080683dd",
"bytes": 3315962,
"r2": 0.8120942160631117,
"mae": 6.225878993242997
},
"stacking_ensemble": {
"display_name": "Stacking Ensemble",
"family": "ensemble",
"algorithm": "RidgeStacking",
"version": "3.0",
"requires_scaling": false,
"file": "models/ensemble/ensemble_stacking.joblib",
"sha256": "bf0fcea2a6195f58738c465b3ff278614f438071c1ddac2f0da29f98b67ad901",
"bytes": 585,
"r2": 0.9259343333721191,
"mae": 2.7872628277726235,
"f1_macro": 0.4949760765550239,
"f1_weighted": 0.8051245669031513,
"tol_2pct": 0.4724137931034483,
"components": [
"xgboost",
"random_forest",
"extra_trees",
"attention_lstm",
"tft"
]
},
"best_ensemble": {
"display_name": "Best Ensemble (XGB+RF+ET+LSTM+TFT)",
"family": "ensemble",
"algorithm": "WeightedAverage",
"version": "3.0",
"requires_scaling": false,
"components": [
"xgboost",
"random_forest",
"extra_trees",
"vanilla_lstm",
"tft"
],
"weights_method": "optimized_l_bfgs_b",
"weights_file": "models/ensemble/ensemble_weights.json",
"file": null,
"r2": 0.9466103276957124,
"mae": 1.438824592772267,
"tol_2pct": 0.903448275862069,
"f1_macro": 0.6923780487804878,
"f1_weighted": 0.9672708158116063,
"weights_sha256": "19006e0c85ee1d8883729eacb2e983808eedccab812faddcf143f997566229fb"
}
},
"auxiliary_artifacts": {
"re_rct_progression": {
"display_name": "Re/Rct Progression Regressors",
"family": "auxiliary",
"algorithm": "LinearRegressionBundle",
"version": "3.0",
"file": "models/classical/re_rct_progression.joblib",
"sha256": "663cb2d1f7ed9db174ed6ff7a9e1c915cc345b6d1119898913cba538fb0ab216",
"bytes": 782
}
},
"scalers": {
"features_standard": "scalers/features_standard.joblib",
"features_minmax": "scalers/features_minmax.joblib"
},
"scaler_checksums": {
"features_standard": "210c6d337aecb114a837e32bc56f4453a3486aad6b6dabb77b7b4ff583eeea3e",
"features_minmax": "6b73bca8cd53ab63215fffcdd8a9abc7071e34076ed63cad5a4d67be9bc8786b"
},
"champion": "gradient_boosting",
"framework": [
"scikit-learn",
"xgboost",
"lightgbm",
"pytorch",
"tensorflow"
],
"training_date": "2026-03-10",
"checksums": {
"models": {
"models/classical/extra_trees.joblib": "9a69db604e6a7e008a063309b3760ab7b949f234a220821bb12e6cf56b37da77",
"models/classical/gradient_boosting.joblib": "17ae16a3244e10d860ed152d16eaa491d75fce5543b3ac9cf60e2a689a45f404",
"models/classical/knn_k5.joblib": "7786c46318c8c5d9c41ad5ab37722289088cacbfda57534cf8450c000a588e8d",
"models/classical/lightgbm.joblib": "2654a3fa80438261bda6b49cb47f7c41c80756810ae6bad3c0e91617ff286fc7",
"models/classical/random_forest.joblib": "dd1535f732486f70ddc881f1a5d48d2ed43735e2bf932cd3d985a1c72b73cc0d",
"models/classical/re_rct_progression.joblib": "663cb2d1f7ed9db174ed6ff7a9e1c915cc345b6d1119898913cba538fb0ab216",
"models/classical/ridge.joblib": "6241bd2746d8fdbb611401f079949491fdb1cfef35c9fa29bae021cecb71090b",
"models/classical/svr.joblib": "8b94d7f86b8adb067870435097bb87e541aeda9baefc35bb8332f1c29d3cc709",
"models/classical/xgboost.joblib": "879f93f0d4dcc126b996c76466907add4edcea0c3c5506566fb364d1fa1ea858",
"models/deep/attention_lstm.pt": "d08b1e02e0328925f0c105398226bacbf1efcf2ccf850a23dab6fd176fc3a9cc",
"models/deep/batterygpt.pt": "f4bcca951f4d32c48a1f731359f999e83f22ae3ac95480d7ef2a084844122587",
"models/deep/bidirectional_lstm.pt": "60aabaf65ebaa709e97e29d7c72e189b7f2157d1a4607d05d461b053cebe5352",
"models/deep/dynamic_graph_itransformer.keras": "a38fa4831fb1c4c840b04d8b1e2372bb54cb9640696ef207dfc59a1165207272",
"models/deep/gru.pt": "38b5af307d2d53cd6c9ca385b8a3f9350d558613fb9f236429902f00a67e012e",
"models/deep/itransformer.keras": "4812ea0ce7757d9f737c8487162f0097d9574ac0d22cc6d032da174fc14d5d8d",
"models/deep/physics_itransformer.keras": "ea915e5091e1f8e7150adddc76ec036c3ff73d56fd1250fb422f42a34c33cba0",
"models/deep/tft.pt": "0c9aa7246ce7974c11d1e441e28b954df0ffd5a189f3692a14ccd24559294abc",
"models/deep/vae_lstm.pt": "a0f0023c697c94bc49e11666e0e0cdcdd92256b6c6d234043834ce27080683dd",
"models/deep/vanilla_lstm.pt": "ffb4717050bebb7c3f2a16528dcf8f909d02f4bc33726abbae6c842ccf66bc4f",
"models/ensemble/ensemble_stacking.joblib": "bf0fcea2a6195f58738c465b3ff278614f438071c1ddac2f0da29f98b67ad901",
"models/ensemble/ensemble_weights.json": "19006e0c85ee1d8883729eacb2e983808eedccab812faddcf143f997566229fb"
},
"scalers": {
"scalers/features_minmax.joblib": "6b73bca8cd53ab63215fffcdd8a9abc7071e34076ed63cad5a4d67be9bc8786b",
"scalers/features_standard.joblib": "210c6d337aecb114a837e32bc56f4453a3486aad6b6dabb77b7b4ff583eeea3e"
},
"results": {
"results/classical_soh_results.csv": "96e4fc1f90de17f373e6501ef64e367882f5aa06ff26715ed0dab5067e2e4fb5",
"results/dg_itransformer_results.json": "a3b71e50dacc03ccd8c134c69096160c0a2b66b1c2fdaacfb87878fd271f5001",
"results/dg_predictions.npz": "f07d344d0c636449723f1ecf7ee74eb49afe2abfb58f8c9210cafd5420f068c1",
"results/ensemble_predictions.npz": "5a1c22492f7a730eb5376d6b02f5bcac58cc6d0253629b611ef144ef66986866",
"results/ensemble_results.csv": "5bddf8f93cee25840a4ba35f27205238ab23302d8595bc43b78e2c3224e67393",
"results/final_rankings.csv": "122ab78f71f415fcc515a9bb8da06effb5c89a4ffddb5e708cae567319aee703",
"results/lstm_predictions.npz": "ae74c989ae87e942f560d27f357d754875647603795eefcc293a701a91cb6283",
"results/lstm_soh_results.csv": "e0e184af259a83248157149c27147930440360a05f7013726d77cf4816b94eb3",
"results/transformer_predictions.npz": "7b73656ba5aebd1714f01f8fe589745de7611ca453746d3b1a6dc2491473c23e",
"results/transformer_soh_results.csv": "dff71aabb993a712e529c95aba6205231cefd91b48b89bf9a238802562d2dc41",
"results/unified_results.csv": "122ab78f71f415fcc515a9bb8da06effb5c89a4ffddb5e708cae567319aee703",
"results/vae_lstm_results.json": "90a7292904dc27e1c57a7100edf7eed1c98d461144c089ed81b543b3d270f44f",
"results/vae_predictions.npz": "144a664a10431cd6c7991204643b101714fa8c6ee8303765b216800049492837"
},
"features": {
"features/battery_features.csv": "56e19c94347af65f60f0cf2b0f241aa926afc8c4b1b97c217db6badfd6ba7f18",
"features/battery_sequences.npz": "55362061ef7f307431695f22e99a046169e2cd5d7b7323cb6d785d1333e43efd",
"features/test_split.csv": "f27b83944869f11128ee049a40c5f30fde17eb29811411b026df2b9272052fdb",
"features/train_split.csv": "f66a5ca940d9457422e9c03907030ccccfdfc5dbeadb0428bbef3df4f0efd65c"
},
"figures": {
"figures/best_ensemble_prediction.png": "061067ebfa0047b8e211120e740646bd34e077105f90f4cbe57201db8ea85682",
"figures/capacity_and_rul.png": "7f959be7175d1ec327791043e21b458e85704efdcd28b4367c85af8189f2464f",
"figures/ced_curves.png": "8a229e96f4c6603b6aef36e9d93da9e5ab69d6a0d37e15307eb588702a7b07ec",
"figures/degradation_state_distribution.png": "cd56c91bee68ac8af3fae38fa3d66191fa1c12c41893792e0c0e7314026d95aa",
"figures/dg_itransformer_predictions.png": "cf429b99f6a04a9e2782dc99605cb55400f8722fcdc3fec8141ed0e9cdc6d230",
"figures/dg_itransformer_training.png": "de1ef81c12b9a66fe40e01ff45cd119e95f70cbf74de868058a4d57610c473af",
"figures/ensemble_comparison.png": "688dc5db69eb58143bddfc69a6b8d5c4b7f0edef2c82af8d9f47f5c77bbca5e9",
"figures/ensemble_confusion_matrix.png": "32cba94422cb6a035c4b14e9a00796e0f28080249df9310f4c92eb71ca6ec81e",
"figures/ensemble_weights.png": "bc4e7bb2f3f28603cc0c87e822705350678bef74e648adebea4fea3a867d9ad3",
"figures/feature_correlation_heatmap.png": "a5ec607126b313e66a70688fa7a56227f4f8360c890479a51033fa226ff77ddf",
"figures/lstm_actual_vs_predicted.png": "da2f66430f05bc90bb3c6196d627a7ab8e5122f5e3b825b584ad58acd7050e2e",
"figures/lstm_training_curves.png": "2449a96970013a4eb356dbb185122a7a1e94339e3f41d75b8e4fc9c560adf10d",
"figures/mc_dropout_uncertainty_lstm.png": "dad1598483196fa7686b77ba89dcd44413f8a3c0ec19c10f09b8aabf61fd6542",
"figures/soc_coulomb_counting_demo.png": "c63c6baca51effb37a89a7a84bf3faa81f52002ff032d1ec7cf0bcdf45bfd15e",
"figures/soh_degradation_trends.png": "56c131fbf98676e8f6611e30bf16a4f6065a311605235c442322a8f71d0fad8e",
"figures/transformer_pt_training_curves.png": "20c4ea6e1802b7557cbf01c439ec51dc4fda230a79a081016fcda4bec8065040",
"figures/transformer_tf_training_curves.png": "a278a36ef3150dfb7fdeb2f89559ac18ccfafd4a4b6ae80d047c0de926e8a9ac",
"figures/unified_model_comparison.png": "5a9de76ea1a4c49327875db1a894afb7c30ea65f4235700726e447c24fdeaa37",
"figures/vae_anomaly_detection.png": "00ae9cb490875ba5ee79dab824619bf4faf1dd6d0a1a113502164d3ba024f412",
"figures/vae_latent_umap.png": "cf17f7a90a0abc9c851b05d5cdbab3f7e34bda0a805fad665a3d1259f3ca7eec",
"figures/vae_lstm_prediction.png": "b41dd8dbe3647a226fb6fb8256b6d528e2a944d64e80ff7d56cc2ae8c875f6cd"
},
"generated_at_utc": "2026-03-10T18:10:44.760892+00:00"
},
"verification": {
"hash_algorithm": "sha256",
"required": true,
"notes": "Verify checksums before serving or deploying artifacts.",
"last_verified_utc": "2026-03-10T18:10:44.760892+00:00"
},
"engineered_features": {
"capacity_retention": "Current capacity / initial capacity ratio",
"cumulative_energy": "Cumulative energy throughput (Wh)",
"dRe_dn": "Rate of change of electrolyte resistance per cycle",
"dRct_dn": "Rate of change of charge-transfer resistance per cycle",
"soh_rolling_mean": "Rolling mean SOH over 5-cycle window",
"voltage_slope": "Slope of voltage curve during discharge"
},
"improvements_over_v2": [
"Cross-battery grouped split eliminates data leakage",
"18 features (6 new physics-informed) vs 12 in v2",
"Proper NaN imputation (ffill/bfill/median vs fillna(0))",
"Optimized hyperparameters for all classical models",
"XGBoost R\u00b2 improved from 0.567 to 0.987"
]
}