Yuheng Shen, Tsinghua University