While caffeine can help short term memory, with TBI it can also increase symptoms but it's hard to know if your problems now are from the TBI, the meds, or the going off caffeine? I'd call and ask your doctor?

http://www1.va.gov/environagents/doc...t-patients.pdf