Weight-aware