Described is a system for transcranial stimulation to improve cognitive function. During operation, the system generates a customized stimulation pattern based on damaged white matter. Further, data is obtained representing natural brain oscillations of a subject. Finally, while the subject is awake, one or more electrodes are activated in phase with the natural brain oscillations and based on the customized stimulation pattern.