Skip to content

Latest commit

 

History

History
62 lines (42 loc) · 2.6 KB

README.md

File metadata and controls

62 lines (42 loc) · 2.6 KB

enhanced-GPDM

We present enhanced Gaussian Process Dynamical Model (EGPDM), a Bayesian method used for capturing high-dimension data's dynamics and transfer learning ability. Our code is based on PyTorch with CUDA supported.

We also provide an original GPDM and its tutorial in IceLab's repo.

TO-DO

  • add EGPDM model
  • update: calculate the original RMSE instead of normalized
  • fix bugs: 3D-trajectory plot; kernel noises
  • fix bugs: cycle limitation in LBFGS

Prerequisites

  • NumPy
  • Pandas
  • Matplotlib
  • SciPy
  • scikit-learn
  • PyTorch

Data preparation

We provide several processed NASA batteries' data (B. Saha and K. Goebel, 2007) for demonstration. The original and whole datasets we used in the paper can be downloaded from: NASA dataset and Oxford dataset.

Train EGPDM

The script train_tran.py is a ready-to-run demo. We demonstrate the GPDM training and testing process along with the transfer learning ability. Please refer to the detailed comments in the code.

Try to setup different hyperparameters to evaluate the model:

  • D: observation space dimension (determined by observation data)
  • Q: desired latent space dimension, empirically $Q << D$
  • dyn_target: full or delta, delta models higher order feature by defining latent points as $x_t - x_{t-1}$

We offer linear, RBF, Matern3 and Matern5 kernel functions in the code. You can custom your desired kernels in self.observationGP_kernel() and self.dynamicGP_kernel() using linear combination of kernels.

Note that our model initializes most of the learnable parameters to $1$. If you want a more random initialization, set the parameters to torch.randn() and use a random seed to control.

License

This project is licensed under the GNU General Public License v3.0.

Acknowledgments

This work partly uses the code from CIGP and CGPDM.

If you find our work useful, please cite:

@article{xing2025enhanced,
  title={Enhanced Gaussian process dynamical modeling for battery health status forecasting},
  author={Xing, WW and Zhang, Z and Shah, AA},
  journal={Renewable and Sustainable Energy Reviews},
  volume={208},
  pages={115045},
  year={2025},
  publisher={Elsevier}
}