sparse_feature_masks -> feature_sets