Work on laying back or sitting ahead of the beat deliberately. Record to a click/loop to refine consistency and pocket.
More in the
Improvisation Practice Systems
category...