I have found that meditation helps with this. Sitting quietly and focusing only on my breath helps retrain my brain to pay attention for longer periods of time. And as the saying goes, if you don't have time to meditate for 30 minutes, you should meditate for an hour.