Key Parameters in the GNN Architecture
The weights of different layers in the GNN are jointly updated with the BKF parameters during training.
| Layer Name | Number of Parameters |
|---|---|
| Linear | 1 × 3 |
| P (covariance matrix) | 3 × 3 |
| Linear (input layer) | 6 × 128 |
| SAGEConv | 128 × 128 |
| BatchNorm1d | 128 × 1 |
| Linear (output layer) | 1 × 128 × 3 |