{ "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" ] }