[Modify] rename top_k_dist into top_k_scores