Yi gra również w reprezentacji Chin, reprezentuje swój kraj na Igrzyskach Olimpijskich w latach 2004, 2008, 2012 i 2016.
, a także Mistrzostwa Świata FIBA 2006 i 2010. Yi zagrał na Igrzyskach Olimpijskich w 2004 roku w wieku 19 lat.