Node-wise Feature Encoding for Neural Performance Prediction
This work introduces FeatureFormer, a neural performance predictor that incorporates explicit node-wise encodings of FLOPs, parameter counts, and memory proxies within a gated graph attention architecture and presents NNEQ, a new large-scale energy consumption dataset that enables unified evaluation of latency and energy prediction.
Matthew Grenier, William Hammer, Andrew Heuer et al.
· 0 citations