Feature aggregation for neural networks