One paper is accepted in AISTATS 2024.