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.