AFAICT, the only problematic widgets are:
vi-backward-delete-char
vi-kill-line
vi-backward-kill-word
So you could do
zle -A kill-whole-line vi-kill-line
zle -A backward-kill-word vi-backward-kill-word
zle -A backward-delete-char vi-backward-delete-char