Two-dimensional (2D) nanochannel membranes stacked by nanosheets are promising membrane separation materials. However, intrinsic swelling properties of 2D membrane have emerged as a significant challenge to establish stable nanochannels and achieve high separation performance. Inspired by the natural ion exchange property of montmorillonite (MMT), we constructed a two-dimensional MMT membrane with robust nanochannels by exchanging Fe3+ into interlayer of the membrane, and further achieved efficient separation of Li+/Mg2+ through coordination manipulation of tannic acid (TA) and Fe3+. Swelling was dramatically prevented by strong interlayer interaction between exchanged Fe3+ and nanosheets resulting in a 32-fold increase in anti-swelling properties. Furthermore, experiments and simulations revealed that TA could rapidly chelate with Fe3+ to form a dense hydrophilic functional layer on the membrane interface, which endowed MMT membranes with enhanced mechanical strength, cation transport and recognition, as well as anti-fouling properties. Consequently, the resulting MMT membranes showed the Li+/Mg2+ selectivity as high as 9.98 with a fast Li+ permeation rate (0.108 mol m-2h−1), which exceeded most of the previously reported membranes. This study proposed a novel strategy to diminish the trade-off effect among stability, ion flux, and selectivity of membranes, and inspired the development of next-generation ion selective separation membranes.