IDEAS home Printed from https://ideas.repec.org/a/eee/appene/v384y2025ics0306261925001497.html
   My bibliography  Save this article

Personalized federated learning for household electricity load prediction with imbalanced historical data

Author

Listed:
  • Zhu, Shibo
  • Shi, Xiaodan
  • Zhao, Huan
  • Chen, Yuntian
  • Zhang, Haoran
  • Song, Xuan
  • Wu, Tianhao
  • Yan, Jinyue

Abstract

Household consumption accounts for about one-third of global electricity. Accurate results of household load prediction would help in energy management at both the building and the grid levels. Data-driven household load prediction methods have shown great advantages and potential in terms of accuracy. However, these methods still face challenges such as limited data for individual households, diversified electricity consumption behaviors, and data privacy concerns. To solve these problems, this paper proposes a personalized federated learning household load prediction framework (PF-HoLo), which allows personal models to learn collectively, leverages multisource data to capture diverse consumption behaviors, and ensures data privacy. In addition, the global encoder model and mutual learning are proposed to enhance the performance of the PF-HoLo framework considering imbalanced residential historical data. Ablation experiments results prove that the PF-HoLo framework could achieve significant improvements, with 13.41% Mean Square Error and 11.33% Mean Absolute Error, compared to traditional federated learning methods.

Suggested Citation

  • Zhu, Shibo & Shi, Xiaodan & Zhao, Huan & Chen, Yuntian & Zhang, Haoran & Song, Xuan & Wu, Tianhao & Yan, Jinyue, 2025. "Personalized federated learning for household electricity load prediction with imbalanced historical data," Applied Energy, Elsevier, vol. 384(C).
  • Handle: RePEc:eee:appene:v:384:y:2025:i:c:s0306261925001497
    DOI: 10.1016/j.apenergy.2025.125419
    as

    Download full text from publisher

    File URL: http://www.sciencedirect.com/science/article/pii/S0306261925001497
    Download Restriction: Full text for ScienceDirect subscribers only

    File URL: https://libkey.io/10.1016/j.apenergy.2025.125419?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    References listed on IDEAS

    as
    1. Zhou, Bin & Li, Wentao & Chan, Ka Wing & Cao, Yijia & Kuang, Yonghong & Liu, Xi & Wang, Xiong, 2016. "Smart home energy management systems: Concept, configurations, and scheduling strategies," Renewable and Sustainable Energy Reviews, Elsevier, vol. 61(C), pages 30-40.
    2. Tang, Lingfeng & Xie, Haipeng & Wang, Xiaoyang & Bie, Zhaohong, 2023. "Privacy-preserving knowledge sharing for few-shot building energy prediction: A federated learning approach," Applied Energy, Elsevier, vol. 337(C).
    3. Wang, Yi & Gan, Dahua & Sun, Mingyang & Zhang, Ning & Lu, Zongxiang & Kang, Chongqing, 2019. "Probabilistic individual load forecasting using pinball loss guided LSTM," Applied Energy, Elsevier, vol. 235(C), pages 10-20.
    4. Li, Ke & Mu, Yuchen & Yang, Fan & Wang, Haiyang & Yan, Yi & Zhang, Chenghui, 2023. "A novel short-term multi-energy load forecasting method for integrated energy system based on feature separation-fusion technology and improved CNN," Applied Energy, Elsevier, vol. 351(C).
    5. Fernández, Joaquín Delgado & Menci, Sergio Potenciano & Lee, Chul Min & Rieger, Alexander & Fridgen, Gilbert, 2022. "Privacy-preserving federated learning for residential short-term load forecasting," Applied Energy, Elsevier, vol. 326(C).
    6. Zhu, Jizhong & Dong, Hanjiang & Zheng, Weiye & Li, Shenglin & Huang, Yanting & Xi, Lei, 2022. "Review and prospect of data-driven techniques for load forecasting in integrated energy systems," Applied Energy, Elsevier, vol. 321(C).
    7. Haider, Haider Tarish & See, Ong Hang & Elmenreich, Wilfried, 2016. "A review of residential demand response of smart grid," Renewable and Sustainable Energy Reviews, Elsevier, vol. 59(C), pages 166-178.
    8. Wan, Anping & Chang, Qing & AL-Bukhaiti, Khalil & He, Jiabo, 2023. "Short-term power load forecasting for combined heat and power using CNN-LSTM enhanced by attention mechanism," Energy, Elsevier, vol. 282(C).
    9. Beckel, Christian & Sadamori, Leyna & Staake, Thorsten & Santini, Silvia, 2014. "Revealing household characteristics from smart meter data," Energy, Elsevier, vol. 78(C), pages 397-410.
    10. Wang, Fei & Lu, Xiaoxing & Chang, Xiqiang & Cao, Xin & Yan, Siqing & Li, Kangping & Duić, Neven & Shafie-khah, Miadreza & Catalão, João P.S., 2022. "Household profile identification for behavioral demand response: A semi-supervised learning approach using smart meter data," Energy, Elsevier, vol. 238(PB).
    11. Tang, Wenjun & Wang, Hao & Lee, Xian-Long & Yang, Hong-Tzer, 2022. "Machine learning approach to uncovering residential energy consumption patterns based on socioeconomic and smart meter data," Energy, Elsevier, vol. 240(C).
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Chen, Haoyu & Huang, Hai & Zheng, Yong & Yang, Bing, 2024. "A load forecasting approach for integrated energy systems based on aggregation hybrid modal decomposition and combined model," Applied Energy, Elsevier, vol. 375(C).
    2. Liu, Yixing & Liu, Bo & Guo, Xiaoyu & Xu, Yiqiao & Ding, Zhengtao, 2023. "Household profile identification for retailers based on personalized federated learning," Energy, Elsevier, vol. 275(C).
    3. Wen, Hanguan & Liu, Xiufeng & Yang, Ming & Lei, Bo & Xu, Cheng & Chen, Zhe, 2024. "A novel approach for identifying customer groups for personalized demand-side management services using household socio-demographic data," Energy, Elsevier, vol. 286(C).
    4. Chen, Bingyang & Zeng, Xingjie & Zhang, Weishan & Fan, Lulu & Cao, Shaohua & Zhou, Jiehan, 2023. "Knowledge sharing-based multi-block federated learning for few-shot oil layer identification," Energy, Elsevier, vol. 283(C).
    5. Ioanna-M. Chatzigeorgiou & Christos Diou & Kyriakos C. Chatzidimitriou & Georgios T. Andreou, 2021. "Demand Response Alert Service Based on Appliance Modeling," Energies, MDPI, vol. 14(10), pages 1-15, May.
    6. Davarzani, Sima & Pisica, Ioana & Taylor, Gareth A. & Munisami, Kevin J., 2021. "Residential Demand Response Strategies and Applications in Active Distribution Network Management," Renewable and Sustainable Energy Reviews, Elsevier, vol. 138(C).
    7. Jieyi Kang & David Reiner, 2021. "Machine Learning on residential electricity consumption: Which households are more responsive to weather?," Working Papers EPRG2113, Energy Policy Research Group, Cambridge Judge Business School, University of Cambridge.
    8. Wang, Danhao & Peng, Daogang & Huang, Dongmei & Zhao, Huirong & Qu, Bogang, 2025. "MMEMformer: A multi-scale memory-enhanced transformer framework for short-term load forecasting in integrated energy systems," Energy, Elsevier, vol. 322(C).
    9. Hu, Rong & Zhou, Kaile & Lu, Xinhui, 2025. "Integrated loads forecasting with absence of crucial factors," Energy, Elsevier, vol. 322(C).
    10. Mohammad Shakeri & Nowshad Amin & Jagadeesh Pasupuleti & Abolfazl Mehbodniya & Nilofar Asim & Sieh Kiong Tiong & Foo Wah Low & Chong Tak Yaw & Nurul Asma Samsudin & Md Rokonuzzaman & Chong Kok Hen & C, 2020. "An Autonomous Home Energy Management System Using Dynamic Priority Strategy in Conventional Homes," Energies, MDPI, vol. 13(13), pages 1-14, June.
    11. Zhang, Le & Zhu, Jizhong & Zhang, Di & Liu, Yun, 2023. "An incremental photovoltaic power prediction method considering concept drift and privacy protection," Applied Energy, Elsevier, vol. 351(C).
    12. Nizami, Sohrab & Tushar, Wayes & Hossain, M.J. & Yuen, Chau & Saha, Tapan & Poor, H. Vincent, 2022. "Transactive energy for low voltage residential networks: A review," Applied Energy, Elsevier, vol. 323(C).
    13. Killian, M. & Zauner, M. & Kozek, M., 2018. "Comprehensive smart home energy management system using mixed-integer quadratic-programming," Applied Energy, Elsevier, vol. 222(C), pages 662-672.
    14. da Fonseca, André L.A. & Chvatal, Karin M.S. & Fernandes, Ricardo A.S., 2021. "Thermal comfort maintenance in demand response programs: A critical review," Renewable and Sustainable Energy Reviews, Elsevier, vol. 141(C).
    15. Khosrowpour, Ardalan & Jain, Rishee K. & Taylor, John E. & Peschiera, Gabriel & Chen, Jiayu & Gulbinas, Rimas, 2018. "A review of occupant energy feedback research: Opportunities for methodological fusion at the intersection of experimentation, analytics, surveys and simulation," Applied Energy, Elsevier, vol. 218(C), pages 304-316.
    16. Kim, Hakpyeong & Choi, Heeju & Kang, Hyuna & An, Jongbaek & Yeom, Seungkeun & Hong, Taehoon, 2021. "A systematic review of the smart energy conservation system: From smart homes to sustainable smart cities," Renewable and Sustainable Energy Reviews, Elsevier, vol. 140(C).
    17. Sana Iqbal & Mohammad Sarfraz & Mohammad Ayyub & Mohd Tariq & Ripon K. Chakrabortty & Michael J. Ryan & Basem Alamri, 2021. "A Comprehensive Review on Residential Demand Side Management Strategies in Smart Grid Environment," Sustainability, MDPI, vol. 13(13), pages 1-23, June.
    18. Hu, Likun & Cao, Yi & Yin, Linfei, 2024. "Long-term price guidance mechanism for integrated energy systems based on gated recurrent unit - vision transformer prediction and fractional-order stochastic dynamic calculus control," Energy, Elsevier, vol. 312(C).
    19. Fan, Pengdan & Wang, Dan & Wang, Wei & Zhang, Xiuyu & Sun, Yuying, 2024. "A novel multi-energy load forecasting method based on building flexibility feature recognition technology and multi-task learning model integrating LSTM," Energy, Elsevier, vol. 308(C).
    20. Lin, Zhengyang & Lin, Tao & Li, Jun & Li, Chen, 2025. "A novel short-term multi-energy load forecasting method for integrated energy system based on two-layer joint modal decomposition and dynamic optimal ensemble learning," Applied Energy, Elsevier, vol. 378(PA).

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:eee:appene:v:384:y:2025:i:c:s0306261925001497. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Catherine Liu (email available below). General contact details of provider: http://www.elsevier.com/wps/find/journaldescription.cws_home/405891/description#description .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.