SA-HGNN: Predicting Power Outages by Teaching Neural Networks About Geography

Paper: 2604.04916 Authors: Xuyang Shen, Zijie Pan, Diego Cerrai, Xinxuan Zhang, Christopher Colorio, Emmanouil N. Anagnostou, Dongjin Song Categories: cs.LG

The Gap

Existing power outage prediction systems treat each location independently, ignoring that storms don’t respect grid boundaries. When a hurricane hits Connecticut, the wind doesn’t stop at town lines—it cascades across regions, knocking out power in spatial patterns. Prior models (like the UConn-Eversource OPM system) use tabular features per location but miss this spatial dependency. They also struggle with class imbalance: hurricanes are rare but devastating, while ice storms are common but localized. The result? Models overfit to frequent events and fail on the rare-but-critical ones.

Problem: Outages spread spatially, events vary in frequency
    |
    v
Assumption: Spatial graph + contrastive learning can capture both
    |
    v
Method: SA-HGNN (spatial encoding) + contrastive loss (event balance)
    |
    v
Evidence: SOTA on 4 territories, handles rare events better
    |
    v
Conclusion: Spatial awareness + event-level embeddings improve prediction

The Increment

One sentence: Before, outage models treated locations as isolated points; now they understand that geography matters and rare events need special attention.

Core Mechanism

SA-HGNN has three stages. First, it builds two graphs: a static graph connecting locations by infrastructure (power lines, land cover) and a dynamic graph connecting them by weather similarity (wind speed, precipitation). Second, it runs separate graph neural networks on each, then fuses their embeddings. Third, it applies contrastive learning: for each weather event type (hurricane, ice storm, etc.), it pulls together embeddings of locations that behave similarly during that event, while pushing apart embeddings across different event types.

Input: Location features (static + dynamic weather)
  |
  v
[Static Graph GNN] ---> Static embeddings
  |                              |
  |                              v
[Dynamic Graph GNN] --> Dynamic embeddings
  |                              |
  +------------------------------+
                |
                v
         [Fusion Layer]
                |
                v
         Fused embeddings
                |
                v
    [Contrastive Learning]
    (Pull similar locations together per event,
     push different events apart)
                |
                v
    Location-specific embeddings
                |
                v
         [Classifier]
                |
                v
    Outage prediction (0-4 levels)

Think of it like a weather forecaster who’s also a city planner. The static graph is the city map—it knows which neighborhoods share power substations, which areas have tree cover that might fall on lines. The dynamic graph is the live weather radar—it sees which zones are getting hit by the same wind gust right now. The forecaster looks at both maps simultaneously, overlaying them to predict where outages will cascade. But here’s the trick: the forecaster has a mental model for each disaster type. When they see a hurricane pattern, they recall “last time this happened, coastal areas went dark first, then inland.” That’s the contrastive learning—building event-specific intuition so rare disasters aren’t treated like common ones.

Key Concepts

• Hybrid Graph Construction: Most GNNs use one graph. This paper uses two because infrastructure and weather operate on different timescales. Static features (power lines, land cover) change slowly; dynamic features (wind, rain) change hourly. By separating them into two graphs, the model can learn different aggregation patterns: static graphs use fixed connectivity (e.g., “these towns share a substation”), while dynamic graphs use weather-based similarity (e.g., “these locations are experiencing similar wind speeds right now”). The fusion layer then combines both perspectives before prediction.

• Contrastive Learning for Event Imbalance: Standard classification treats all training samples equally. But if you have 1000 ice storm examples and 10 hurricane examples, the model learns to predict ice storms well and ignores hurricanes. Contrastive learning fixes this by creating a separate embedding space per event type. During training, for each event (e.g., Hurricane Sandy), it pulls together embeddings of all locations affected by that event, regardless of how many samples exist. This forces the model to learn event-specific patterns even from rare occurrences. It’s like giving each disaster type its own chapter in the model’s memory, so rare events aren’t drowned out by common ones.

• Spatially Aware Aggregation: Traditional GNNs aggregate neighbor features uniformly. SA-HGNN weights neighbors by spatial distance and feature similarity. If two locations are 50 miles apart but experiencing identical wind speeds, they should influence each other more than two locations 5 miles apart with different weather. The model learns these weights during training, effectively discovering which spatial scales matter for outage propagation (e.g., hurricanes affect 100+ mile radii, ice storms affect 10-mile radii).

Framework Shift

Before (mainstream approach):        After (this paper):

Location A --> [Features] --> Model  Location A ---+
Location B --> [Features] --> Model               |
Location C --> [Features] --> Model      [Static Graph GNN]
(Independent predictions)                         |
                                                   v
                                          Static embeddings
                                                   |
                                                   +---> [Fusion]
                                                   |         |
                                          Dynamic embeddings v
                                                   ^    [Contrastive]
                                                   |         |
                                      [Dynamic Graph GNN]   v
                                                   |    Predictions
Weather at A ---+                                  |
Weather at B ---+                                  |
Weather at C ---+                                  |
(Spatial relationships ignored)    (Spatial + event-aware)

From isolated feature vectors to spatially-connected graphs with event-specific embeddings, the core shift is treating outage prediction as a spatial propagation problem rather than a per-location classification task.

Expert Assessment

Problem choice: Real gap. Existing OPM systems are deployed in production but genuinely miss spatial effects—this isn’t academic navel-gazing. The UConn-Eversource partnership gives access to actual utility data, which is rare in this domain. The problem sits at the intersection of graph learning and imbalanced classification, both active areas.

Method maturity: Solid engineering, not groundbreaking theory. Hybrid graphs are known (e.g., heterogeneous GNNs), contrastive learning for imbalance is established (e.g., supervised contrastive loss). The novelty is in the combination and domain application. The spatial weighting mechanism is clever but not deeply analyzed—why this aggregation function over others? The paper doesn’t ablate design choices rigorously.

Experimental integrity: Baselines are fair (XGBoost, standard GNN, temporal models). Four territories provide good generalization evidence. However, the evaluation metric (macro F1) hides class-specific performance—I’d want to see per-event-type confusion matrices. The contrastive learning ablation (Table 3) shows clear gains, but the paper doesn’t explain *why it works better for rare events beyond intuition. Also, no discussion of computational cost—GNNs on large graphs can be expensive for real-time prediction.

Writing quality: The method section is dense and assumes GNN familiarity. Figure 2 (architecture diagram) is cluttered—splitting it into separate static/dynamic/fusion diagrams would help. The related work section is thorough but reads like a literature dump. The results section is strong, with good visualizations (Figure 4’s spatial heatmaps are excellent). If I could rewrite one section, it’d be the method—walk through a concrete example (e.g., “Hurricane Sandy hits location X, here’s how SA-HGNN processes it step-by-step”) before diving into formalism.

Verdict: weak accept — Solid applied work with real-world impact, but incremental methodologically. The spatial awareness is the key contribution; contrastive learning is a known trick applied competently. Worth publishing for the domain contribution and empirical validation, but not a methodological leap.

Takeaways

Hybrid graphs for multi-timescale data: If your problem has both slow-changing structure (e.g., road networks, social connections) and fast-changing signals (e.g., traffic, messages), don’t force them into one graph. Build separate graphs, run separate GNNs, fuse late. This applies beyond power grids—think epidemic modeling (static: contact networks, dynamic: mobility patterns) or financial networks (static: ownership, dynamic: transaction flows).

Contrastive learning as a class balancing tool: Instead of reweighting loss functions or oversampling rare classes, use contrastive learning to create event-specific embedding spaces. This is especially useful when rare classes have distinct patterns (e.g., hurricanes vs. ice storms) rather than just being underrepresented versions of common classes. The key: define positive pairs within each class, negative pairs across classes.

Spatial weighting in GNNs: Don’t assume uniform neighbor influence. Learn attention weights based on both distance and feature similarity. This is underused in GNN applications—most papers either use fixed adjacency or pure attention without spatial priors. The trick: initialize attention with distance-based bias, then let the model refine it.