Localize weight based on KG