SOURCE-LINKED INTELLIGENCE

Node-wise Feature Encoding for Neural Performance Prediction

arXiv · AI, language, vision and robotics · article · Aug 28, 2026 · UTC

As neural networks are increasingly deployed on resource constrained edge devices, accurate prediction of latency and energy is critical for efficient neural architecture search. Existing GNN and transformer based predictors achieve strong results but largely ignore node-level computational cost, limiting their ability to model performance critical operations. To address this, we introduce FeatureFormer, a neural performance predictor that incorporates explicit node-wise encodings of FLOPs, parameter counts, and memory proxies within a gated graph attention architecture. We also present NNEQ,

Read original source ↗ Open in workspace

recordType

paper

region

Global