If broker sense that it is unhealthy, broker should remove itself as leader from all partitions that it is hosting as partition leader.

A broker can have partition leader for more than one partition.

If broker sense any panic with the host like disk write, memory (crashing), network. 

It should send eject me ( like ejecting from fighter jet) from cluster message to cluster controller/zookeeper. Which means its not a part of cluster anymore.

This makes the next leader election quicker and can prevent many issues before it happens.