revise output interfaces for aggregation training