Yi speelt ook voor het Chinese nationale team, en vertegenwoordigde zijn land op de Olympische Spelen in 2004, 2008, 2012.
en 2016, evenals de FIBA Wereldkampioenschappen 2006 en 2010. Yi speelde op 19-jarige leeftijd in 2004 op de Olympische Spelen.