Treatment for kidney disease depends on the underlying cause and severity. For acute kidney injury, removing the source of toxicity and providing supportive care can often restore kidney function. Chronic kidney disease may require more extensive management, including medication to control symptoms, dietary changes, and in severe cases, dialysis or a kidney transplant.