Kee has signed a new three-year contract with the League Two outfit.
"I've really enjoyed my time here so far so I'm delighted to have signed a new contract," Kee told the club's official website.
"I've settled here - we're like a big, happy family - and you can see that with the way my football has come on in the last year. I did well last season and now the club have rewarded me for getting back fit."
Albion boss Gary Rowett added: "Billy was coming into his last year and, after his goal exploits last year and his age and international prospects, I felt as if he was somebody we definitely wanted to tie down for the longer term.
"I'm really pleased that, after three of four weeks of talking to him, it's come to fruition. Billy has worked really hard over the summer. He's shown a real commitment to the cause. He's expressed his enjoyment to be at this club and, if he's enjoying his football and his environment, he'll score goals."