Heterogeneity-aware personalised federated learning for household energy forecasting on multi-source real-world smart meter data
摘要
Residential and building-level electricity forecasting is increasingly based on smart-meter streams, but direct data pooling is often difficult because load traces are private, geographically dispersed and statistically different across households and buildings. This paper studies the effect of such client heterogeneity in federated energy forecasting. A multi-source benchmark is constructed by combining processed Smart, CU-BEMS, UCI Household and AMPds clients, giving 22 usable clients and 1,522,510 observations at 15-minute resolution. The benchmark compares non-federated baselines (Persistence, Ridge, LocalOnly and Centralised training), standard federated baselines (FedAvg, FedProx and FedPer) and the proposed HAPFL framework over 1-step, 12-step and 24-step forecasting horizons. HAPFL separates a shared temporal encoder from client-specific prediction heads and combines proximal stabilisation, latent prototype alignment and difficulty-aware aggregation. In the reported benchmark protocol, HAPFL gives the lowest mean MAE and the highest mean