You can modify undo to chunk based on words with the following:

:inoremap <Space> <Space><C-g>u