The 26-year-old came through the ranks with Hearts and made his first-team debut in 2012 before heading south to join the Latics in January 2018.
Walker now returns to the capital as Craig Levein's third signing of the summer after Craig Halkett (Livingston) and Conor Washington (Sheffield United).