50 оттенков линейной регрессии, или почему всё, что вы знаете об A-B тестах, помещается в одно уравнение

Всем привет! A/B тестирование уже давно стало стандартом в проверке гипотез и улучшении продуктов в X5. Но, как ни странно, многие из «модных» техник, которые применяются в A/B тестировании, на самом деле, не что иное, как вариации старой доброй линейной регрессии. 

Например, использование таких методов, как t-тест, стратификация, CUPED, CUMPED, по сути, сводятся к построению линейной регрессии и проверке гипотезы в рамках построенной модели. Наши коллеги из команды ad-hoc аналитики Х5 Tech уже писали про стратификацию здесь и про CUPED здесь, а по этой ссылке лежит статья про CUMPED и подобные вариации CUPED.

Основная идея здесь проста: правильное добавление новых переменных в модель помогает лучше контролировать внешние факторы и уменьшать шум в данных. Это позволяет точнее оценить эффект от воздействия и объединить разные статистические подходы, которые обычно рассматриваются отдельно. Но почему это работает? Почему всё сводится к тому, что добавление переменных помогает объединить, казалось бы, разрозненные техники? 

Чтобы разобраться в этом, для начала вспомним основы линейной регрессии, после чего перейдём к различным статистическим методам снижения дисперсии и покажем, как они сводятся к линейной регрессии. Затем объединим все техники вместе и на примере покажем, как они работают на практике.

Парная регрессия

Линейная регрессия моделирует взаимосвязь между независимыми и зависимой переменными, подбирая прямую линию (или гиперплоскость в случае множества переменных), которая лучше всего описывает зависимость между ними. Независимая переменная – это переменная, которую мы считаем влияющей на зависимую переменную. Зависимая переменная – интересующая нас переменная, влияние на которую мы хотим изучить. Независимые переменные объясняют изменения зависимой переменной. Уравнение линейной регрессии для случая с одной независимой переменной  выглядит следующим образом:

где – зависимая переменная;– независимая переменная;– свободный член, значение точки пересечения оси ординат;– это коэффициент наклона, показывающий изменениепри изменениина единицу, при условии, что все другие независимые переменные остаются неизменными;— случайная ошибка, которая учитывает влияние всех других факторов, которые не были включены в модель.

Например, предположим, что мы хотим изучить, как количество дней с активными промоакциями влияет на недельную выручку в магазине. У нас есть данные, собранные из нескольких магазинов. Для каждого магазина в наборе данных есть два значения:– количество дней с активными промоакциями и– недельная выручка. В данном случае количество дней с активными промоакциями будет независимой переменной, а недельная выручка будет зависимой переменной.

Для нахождения наиболее точной модели, описывающей взаимосвязь между переменными, чаще всего используется метод наименьших квадратов, он минимизирует сумму квадратов отклонений между реальными значениямии предсказанными. Сумма квадратов отклонений называется RSS (Residual Sum of Squares).

По сути, метод наименьших квадратов помогает найти такую линию, которая проходит как можно ближе к большинству точек на графике.

Перепишем уравнение RSS:

Найдём частные производные для​и и приравняем их к нулю. Решив систему уравнений, получим:

Упростив, получим:

Где — ковариация междуи;— дисперсия.

Для каждого коэффициента мы можем оценить его статистическую значимость. Это делается с помощью t-статистики и p-значений. Если p-значение меньше определённого уровня значимости (например, 0,05), мы можем сказать, что коэффициент статистически значим. Это означает, что при условии, что на самом деле изменение независимой переменной не оказывает влияния на зависимую переменную, вероятность получения такой t статистики или более экстремального значения (по модулю) крайне мала. Таким образом, мы можем сделать вывод, что изменение независимой переменной, скорее всего, оказывает влияние на зависимую переменную.

При определённых предположениях можно доказать (что выходит за рамки данной статьи), что оценка коэффициента сходится по распределению к нормальному распределению, и поэтому мы можем построить доверительный интервал:

Где – квантиль нормального распределения, – уровень значимости.

Оценка дисперсии оценки коэффициента равна:

Где— оценка дисперсии ошибок модели (остатков),— значения независимой переменной, — среднее значение независимой переменной.

 Множественная регрессия

Модель линейной регрессии с несколькими переменными записывается следующим образом:

Где — зависимая переменная;​— свободный член,— коэффициенты при независимых переменных;— независимые переменные;— ошибка модели.

Оценка коэффициента будет равна:

Где— это разница между фактическим значением и тем значением, которое мы получили, построив множественную регрессию на независимые переменные, кроме, и константу​.нельзя предсказать с помощью переменных​, которые мы использовали для предсказания, а это значит, что– это часть результата, которая не объясняется этими переменными. 

Аналогично для этого коэффициента мы можем построить доверительный интервал:

Оценка дисперсии оценки интересующего коэффициента будет равна:

— коэффициент детерминации, квадрат коэффициента корреляции между переменной и оставшимися независимыми переменными, присутствующими в модели.

ATE – Average Treatment Effect

Прежде чем углубляться в анализ результатов регрессии, рассмотрим, что такое Average Treatment Effect (ATE), и почему он играет ключевую роль в нашей оценке. ATE — это средний эффект от воздействия. Он оценивается как разница в средних значениях интересующей метрики между двумя группами: тестовой и контрольной. Тестовая группа подвергается воздействию или какому-то изменению, а контрольная группа остается без изменений, чтобы мы могли сравнить результаты в этих группах. Следует отметить, что далее мы рассматриваем исключительно случай рандомизированного эксперимента, где попадание в тестовую и контрольную группы происходит случайным образом. Например, мы тестируем новую выкладку товаров в сети офлайн-магазинов и хотим понять, как это повлияет на объем продаж. В тестовой группе магазинов товары размещаются по-новому, в то время как в контрольной группе выкладка остается прежней. ATE в этом случае покажет, насколько средний объем продаж отличается между магазинами с новой и старой выкладкой.

Формально, ATE можно записать как, где – статус принадлежности к тестовой или контрольной группе (тритмент), а - интересующая метрика. Фактически, в t-тесте мы тоже оцениваем ATE – разность средних значений метрики между двумя группами.

Когда мы оцениваем коэффициенты модели линейной регрессии, в которой переменная, обозначающая принадлежность к тестовой или контрольной группе, представлена бинарной переменной (0 для контрольной группы и 1 для тестовой), оценка коэффициента при этой переменной фактически является оценкой ATE.

Взяв математическое ожидание от этого уравнения в двух случаях (), при определённых предположениях, о которых поговорим позже, получим:

То есть (математическое ожидание метрики в контрольной группе) и(математическое ожидание метрики в тестовой группе). Тогда

Хорошие переменные

Точно так же, как и в парной регрессии, мы можем получить ATE, используя множественную линейную регрессию.

где – зависимая переменная​ – независимые переменные; – свободный член; — коэффициент воздействия при, как раз отражающий ATE; – статус принадлежности к тестовой или контрольной группе​,​ — коэффициенты регрессии, показывающие, насколько изменится при изменении соответствующей переменнойна одну единицу, если остальные переменные остаются неизменными; — случайная ошибка, которая учитывает влияние всех других факторов, которые не были включены в модель.

Основное преимущество использования нескольких переменных заключается в возможности контролировать различные факторы, влияющие на интересующую метрику, что позволяет уменьшить дисперсию оценки коэффициента.

Представим, что у нас есть ковариата(например, размер магазина), которая хорошо предсказывает зависимую переменную(например, объём продаж), но независима от тритмента(например, новая рекламная кампания, которую мы хотим запустить), что будет достигаться в нашем случае, так как мы говорим о рандомизированном эксперименте, то есть магазины выбираются случайным образом в тестовую и контрольную группы. Когда мы добавляем ковариатув модель, она помогает объяснить часть вариации в, которая раньше считалась случайной ошибкой, как бы очищая взаимосвязь междуи. Это приводит к снижению дисперсии, которая стоит в формуле дисперсии оценки коэффициента. Из-за этого дисперсия оценки коэффициента становится меньше.

Чем меньше дисперсия, тем меньше и стандартная ошибка оценки коэффициента

Наши доверительные интервалы становятся уже, что даёт нам возможность обнаруживать более мелкие по размеру эффекты

Где – квантиль нормального распределения, – уровень значимости.

Таким образом, стоит включать переменные, которые являются хорошими предикторами интересующей метрики.

Предпосылки применения линейной регрессии для получения несмещённой и состоятельной оценки эффекта в случае рандомизированного эксперимента.

В контексте машинного обучения при использовании линейной регрессии нам важнее всего наиболее точно спрогнозироватьи в целом нам не очень важны коэффициенты модели и их статистические свойства. Однако в контексте A/B тестирования нас интересует лишь коэффициент при переменной, которая обозначает статус принадлежности к тестовой или контрольной группе (тритмент), так как именно этот коэффициент и показывает эффект от воздействия. Соответственно, нам важно получить качественную оценку коэффициента, то есть оценку несмещённую и состоятельную. Как этого достичь? 

Для начала скажем, что в нашей статье мы предполагаем гомоскедантичность случайных ошибок, то есть постоянство дисперсии случайных ошибок для всех наблюдений. Однако это предположение является не обязательным условием, оно лишь упростит наши формулы для линейной регрессии, которые мы будем использовать далее. А теперь перейдём к предпосылкам:

1. Чтобы у нас была возможность получить оценку интересующего коэффициента, нам необходимо отсутствие строгой мультиколлинеарности. Когда в модели возникает строгая мультиколлинеарность, одна или несколько независимых переменных являются точными линейными комбинациями других. Это означает, что столбцы матрицы признаков линейно зависимы. При наличии такой зависимости матрица, которую мы используем для вычисления коэффициентов регрессии, становится вырожденной. В результате такая матрица не может быть обращена, что необходимо для нахождения коэффициентов регрессии методом наименьших квадратов (формула для оценки коэффициента линейной регрессии методом наименьших квадратов в матричной форме):

2.иимеют ненулевые конечные четвертые моменты распределения.

По существу, это значит, что вероятность появления экстремально больших выбросов в данных крайне мала. Это предположение считается достаточно слабым, поэтому предполагается, что обычно оно выполнимо на практике. 

3. Экзогенность тритмента. Также в контексте нашего анализа важно, чтобы переменная тритментаоставалась экзогенной, то есть условная ковариация переменнойи ошибки модели, при фиксированных переменных, равна нулю:

Докажем несмещённость и состоятельность нашей оценки коэффициента при экзогенности тритмента: 

воспользуемся утверждением без доказательства, что если наблюдения независимы и одинаково распределены, то

Тогда:

Таким образом получаем:

Учитывая, что наша переменная экзогенна, а значит ковариация и ошибки модели равна 0, мы получим состоятельную и несмещённую оценку.

Нарушение предпосылки об экзогенности переменной – одна из самых важных проблем. Давайте рассмотрим возможные причины её возникновения в контексте оценки экспериментов.

Причина 1. Пропуск существенной переменной.

Предположим, мы изучаем эффект новой программы лояльности на продажи магазинов, но не учитываем маркетинговую активность, которая также влияет на продажи. Если магазины с более высокой маркетинговой активностью попали в тестовую группу, а с более низкой – в контрольную, это приведёт к тому, что влияние программы лояльности будет переоценено, так как рост продаж может быть связан не только с новой программой, но и с маркетингом. В таком случае маркетинговая активность является пропущенной переменной, которая вызывает эндогенность интересующей нас переменной.

Однако в случае рандомизированного эксперимента, где магазины случайным образом распределены в тестовую и контрольную группы, пропуск переменной не создаст никакого смещения. Это связано с тем, что влияние маркетинговой активности распределено случайно между тестовой и контрольной группами. Благодаря этому её влияние на интересующую метрику будет одинаковым как в тестовой, так и в контрольной группе, и она не приведёт к возникновению эндогенности. 

Причина 2. Влияние зависимой переменной на ковариаты во время эксперимента.

Предположим, мы изучаем эффект рекламной кампании (тритмент) на количество продаж (зависимая переменная). Одновременно мы добавляем в модель ковариату — количество положительных отзывов о продукте, собранных во время эксперимента, — с целью снижения дисперсии оценки эффекта.

Проблема возникает, если наша зависимая переменная может оказывать влияние на ковариату во время эксперимента. Например, успешная рекламная кампания может увеличить количество продаж, что приведёт к большему числу положительных отзывов.

Тут есть два приемлемых варианта – если мы выбираем использовать ковариату, собранную во время эксперимента, то зависимая переменная никак не должна влиять на эту ковариату. Это может быть пол, возраст и тому подобное. Второй вариант – использовать ковариату, собранную до проведения эксперимента. Таким образом наша зависимая переменная во время эксперимента никак не повлияет на эту ковариату.

Ключевая задача в этом случае — корректно подбирать ковариаты таким образом, чтобы они не подвергались влиянию зависимой переменной во время эксперимента.

Техники A/B тестирования и линейная регрессия

T-Test

Тест Стьюдента — это статистический тест, используемый для сравнения средних значений интересующей метрики между двумя группами. Он помогает определить, значима ли разница между средними значениями тестовой и контрольной групп. Более подробную информацию о t-тесте можно найти здесь. 

Как было показано ранее, коэффициентпри переменной тритмента в линейной регрессии представляет собой разницу средних между тестовой и контрольной группами, которая также является оценкой эффекта (ATE). В формулировке t-теста это выражается как:

t-статистика в t-тесте, оценивающая разницу между средними, имеет вид:

В данном контексте t-test применяется при допущении, что дисперсии между группами равны. Если же это допущение не выполняется, тогда стоит использовать, например, t-тест Уэлча, который не требует равенства дисперсий между группами. В контексте линейной регрессии в этом случае применяются робастные стандартные ошибки, о них можно подробнее прочитать по этой ссылке. 

Продолжим. Оба метода оценивают одну и ту же величину — разницу между средними.

Доверительный интервал в таком случае имеет вид:

Аналогичный доверительный интервал для коэффициента​в линейной регрессии:

Отметим, что поскольку мы работаем с большими данными, t-распределение будет совпадать с нормальным распределением. При достаточно больших выборках оба метода будут давать одинаковые результаты.

Получается, что t-тест эквивалентен построению линейной регрессии, где зависимой переменной является интересующая метрика, а независимой переменной — принадлежность объекта к тестовой группе. В случае линейной регрессии уравнение имеет вид:

где— статус присвоения к тестовой группе (1 для тестовой группы и 0 для контрольной группы),— среднее значениев контрольной группе,— коэффициент, который показывает разницу между средними значениями в тестовой и контрольной группах (ATE),— случайная ошибка.

Таким образом, мы можем сравнить группы в рамках эксперимента как напрямую с помощью t-test, так и путем построения линейной регрессии и проверки данной гипотезы в рамках построенной модели. 

Однако на этом наши возможности не заканчиваются, мы можем улучшить нашу оценку путём добавления новых переменных в регрессию. Поскольку мы рассматриваем случай честного рандомизированного эксперимента, при добавлении переменных в модель речь идёт прежде всего о ковариатах. Как мы показали ранее, добавление ковариат помогает объяснить часть вариации в зависимой переменной, тем самым уменьшая остаточную ошибку модели. А это приводит к уменьшению дисперсии оценок эффекта воздействия, снижению стандартных ошибок и, как следствие, увеличению мощности теста.

Разберём методы, которые помогут нам реализовать этот подход и покажем, что все эти техники можно свести к линейной регрессии.

Стратификация

Наши коллеги уже писали про стратификацию здесь. Если кратко, то стратификация — это метод, который используется для уменьшения дисперсии оценок путём учёта принадлежности объектов эксперимента к определённым стратам, предположительно влияющих на зависимую переменную. Например, это может быть пол, возраст, география и тому подобное. В зависимости от способа формирования выборок и получения оценок стратификация может быть нескольких видов. В контексте линейной регрессии применяется постстратификация, которая используется для коррекции оценок после того, как эксперимент был уже проведён. 

Представим, что мы провели эксперимент, в котором хотели изучить влияние нового продукта. У нас есть данные о возрасте участников эксперимента, и мы хотим учесть этот фактор для уменьшения дисперсии оценок. Для этого мы делим участников на три возрастные группы: молодые, среднего возраста и старшие.

Постстратификация реализуется путём добавления фиктивных переменных в уравнение линейной регрессии. Эти переменные указывают принадлежность наблюдения к одной из возрастных страт и принимают значения 0 или 1. Например, фиктивная переменная , равная 1, будет означать принадлежность объекта к молодой возрастной группе.

 Тогда уравнение линейной регрессии для стратификации будет выглядеть так:

где— зависимая переменная,— статус присвоения к тестовой группе (1 для тестовой группы и 0 для контрольной группы),— фиктивная переменная, обозначающая принадлежность к молодой возрастной группе,— фиктивная переменная, обозначающая принадлежность к старшей возрастной группе,— случайная ошибка.

Важно отметить, что добавляются только две фиктивные переменные (и ), чтобы избежать ловушки дамми-переменных, которая может привести к строгой мультиколлинеарности, если добавить переменные для всех возрастных групп.

Докажем, что значение коэффициента при тритменте в линейной регрессии действительно равно оценке эффекта АТЕ, полученному с помощью постстратификации. 

Имеем:

Где— интересующая метрика,— переменная тритмента (0 или 1),​— фиктивная переменная, которая указывает принадлежность к страте,— коэффициенты для страты, — ошибка модели.

Для каждой страты k можно записать математическое ожидание результата y для контрольнойи тестовойгрупп:

Разность средних в каждой страте k между тестом и контролем будет:

Для вычисления общего эффекта тритмента по всем стратам, используется взвешенная разность средних, где вес каждой страты ​ пропорционален размеру страты:

Так как для каждой страты, то взвешенная разность средних равна:

CUPED (Controlled-experiment Using Pre-Existing Data)

CUPED — это метод, который помогает снизить дисперсию оценок, используя дополнительную информацию, доступную до проведения эксперимента. Основная идея заключается в том, чтобы скорректировать целевую метрику, учитывая предэкспериментальные данные, и затем применить к ней статистический критерий (например, t-test). Таким образом, CUPED устраняет часть вариабельности этой метрики, которую можно объяснить историческими данными, что позволяет более точно оценить эффект от воздействия. Обычно в качестве таких данных используют значения той же метрики, измеренные до начала эксперимента. 

Для расчёта метрики методом CUPED используется следующее уравнение:

где– Среднее значение целевой метрики,– среднее значение ковариаты (метрики до проведения эксперимента), а– коэффициент при, скалярная величина. 

Важно отметить, что принцип CUPED можно обобщить на другие похожие методы, например, CUPAC (Control Using Predictions As Covariate), где ковариата — это наша метрика, предсказанная ML моделью, или CUMPED – Multi CUPED, в котором мы используем не одну, а сразу несколько ковариат. Независимо от метода, добавление качественных ковариат способствует снижению дисперсии оценки. 

Дисперсияравна:

Дисперсияминимизируется при:

Эта формула уже встречалась нам ранее как коэффициент линейной регрессии. Подставивв формулу дисперсии, мы получаем

где— это корреляция междуи.

Таким образом, чем выше корреляция междуи, тем большее уменьшение дисперсии с помощью CUPED.

Покажем, что оценка ATE методом CUPED равна оценке коэффициента линейной регрессии при тритменте beta. Уравнение линейной регрессии будет выглядеть следующим образом:

Оценка ATE методом CUPED:

Где T и C обозначают принадлежность к тестовой или контрольной группе. Математические ожидания сокращаются, ведь они рассчитываются для всей популяции. 

Раскроем скобки:

Найдём разницу между средними значениями метрики в тестовой и контрольной группах в линейной регрессии:

Учитывая, что коэффициентравен, мы получили равенство оценки ATE методом CUPED и оценке коэффициента линейной регрессии.

А если мы сможем найти больше переменных, которые удовлетворяют требованиям экзогенности тритмента, которые были описаны выше? Тогда используем модель линейной регрессии со всеми ковариатами – в нашем случае это будет Multi CUPED (при этом стандартная ошибка оценки коэффициента станет ещё ниже):

Подробнее про то, какие переменные стоит добавлять в модель, а какие не стоит, можно прочитать в этой статье.

Все методы в одном уравнении

Мы рассмотрели несколько методов для снижения дисперсии. Каждый из этих подходов, снижая дисперсию, давал нам возможность получить более точные оценки. Но что, если объединить все эти методы в единое уравнение линейной регрессии? 

Предположим, что мы захотели провести эксперимент, и у нас есть данные о возрасте участников, значении метрики до эксперимента и значения других ковариат, влияющих на интересующую нас метрику. Мы уже рассмотрели, как каждый метод по отдельности, используя различные ковариаты, помогает снижать дисперсию оценки коэффициента, и теперь пришло время объединить их все в одно уравнение:

Объединяя все методы в одном уравнении линейной регрессии, мы одновременно используем все доступные нам ковариаты, таким образом минимизируя дисперсию оценки эффекта на сколько это возможно в нашем случае. 

Получается, что вместо того, чтобы применять методы поодиночке, мы можем интегрировать их в одно общее уравнение линейной регрессии, получив таким образом множественную линейную регрессию с переменной тритмента и всеми возможными ковариатами. 

В итоге уравнение регрессии на случай с множеством ковариат сводится к общему виду:

— зависимая переменная (интересующая метрика),​— свободный член, — тритмент (0 — контрольная группа, 1 — экспериментальная группа),— ковариаты, — коэффициенты при ковариатах,— случайная ошибка модели.

Применение линейной регрессии на практике

Теперь продемонстрируем это всё на практике и посмотрим на стандартные ошибки оценки коэффициента при тритменте. Для начала сгенерируем данные:

def generate_experiment_data(
    n_samples: int = 30_000,
    mean1: int = 30,
    mean2: int = 30,
    std_dev1: int = 4,
    std_dev2: int = 4,
    seed: int = 2003,
) -> pd.DataFrame:
    """Создает экспериментальные данные для контрольной
    и пилотной групп с учетом нескольких ковариат.

    n_samples - количество наблюдений в каждой группе
    mean1, mean2 - средние значения метрик для контрольной и пилотной групп
    std_dev1, std_dev2 - стандартные отклонения метрик для групп

    return - DataFrame с данными эксперимента
    """
    np.random.seed(seed)
    
    # Генерация данных для контрольной и пилотной групп
    group1 = np.random.normal(loc=mean1, scale=std_dev1, size=n_samples)
    group2 = np.random.normal(loc=mean2, scale=std_dev2, size=n_samples)

    # Стратификация
    strata = np.random.choice(["young", "middle", "old"],
                              size=n_samples * 2, p=[0.3, 0.5, 0.2])

    # Дополнительные ковариаты
    covariate1 = np.random.normal(5, 1, size=n_samples * 2)  
    covariate2 = np.random.normal(10, 2, size=n_samples * 2) 
    covariate3 = np.random.normal(20, 5, size=n_samples * 2)  

    #  baseline_metric
    baseline_metric = np.concatenate([group1, group2])

    # пост-метрика
    y_post_metric = (
        baseline_metric * 1.7
        + np.concatenate([np.zeros(n_samples), np.ones(n_samples)]) * 5
        + covariate1 * 0.1  
        + covariate2 * 0.3  
        + covariate3 * 0.5
        + np.random.normal(0, 1.5, size=n_samples * 2)
    )
    y_post_metric += np.where(
        strata == "young", 1,
        np.where(strata == "middle", 5, 10)
    )

    data = pd.DataFrame({
        "baseline_metric": baseline_metric,
        "D": np.concatenate([np.zeros(n_samples), np.ones(n_samples)]),
        "strata": strata,
        "covariate1": covariate1,
        "covariate2": covariate2,
        "covariate3": covariate3,
     "y_post_metric": y_post_metric
    })

    return data


def analyze_experiment(data: pd.DataFrame) -> float:
    """построение графиков плотностей двух групп
    и вычисление разницы средних значений метрик

    data - DataFrame с данными эксперимента

    return - разница средних y_post_metric
    между пилотной и контрольной группами.
    """
    mean_difference = (
        data.loc[data["D"]== 1, "y_post_metric"].mean()
        - data.loc[data["D"] == 0, "y_post_metric"].mean()
    )

    plt.figure(figsize=(10, 6))
    sns.kdeplot(data=data, x="y_post_metric", hue="D", fill=True)
    plt.title("График плотностей")
    plt.xlabel("Metric")
    plt.ylabel("Density")
    plt.show()

    return mean_difference


data = generate_experiment_data()

mean_diff = analyze_experiment(data)
print("Разница средних: ", mean_diff)

Разница средних:  5.0461

Построим линейную регрессию, чтобы проверить гипотезу о равенстве средних с помощью t-test в рамках построенной модели, и посмотрим на стандартное отклонение оценки коэффициента при– нашем эффекте воздействия: 

def perform_ols_regression(data: pd.DataFrame):
    """выполняет OLS регрессию метрики y_post_metric на переменную D
    и возвращает информацию о коэффициентах модели

    data - DataFrame с данными эксперимента

    return - таблица коэффициентов для модели
    """
    model = smf.ols("y_post_metric ~ D", data=data).fit()

    coefficients_table = model.summary().tables[1]

    return coefficients_table

ols_result = perform_ols_regression(data)
print(ols_result)

В таком случае мы получили стандартную ошибку оценки коэффициента при D равную 0.066.

Воспользуемся методом постстратификации с помощью линейной регрессии и посмотрим на результат:

def perform_stratified_regression(data: pd.DataFrame):
    """выполняет регрессию с учетом страт на переменную D
     и возвращает информацию по коэффициентам

    data - DataFrame с данными эксперимента

    return - таблица коэффициентов о модели
    """
    strata_dummies = pd.get_dummies(data["strata"], drop_first=True)
    data_with_dummies = pd.concat([data, strata_dummies], axis=1)
    
    X_strat = sm.add_constant(
        data_with_dummies[["D"] + list(strata_dummies.columns)]
    ).astype(float)

    model_strat = sm.OLS(data["y_post_metric"], X_strat).fit()

    return model_strat.summary().tables[1]


stratified_regression_result = perform_stratified_regression(data)
print(stratified_regression_result)

Стандартное отклонение оценки коэффициента при уменьшилось до 0.060!

Теперь, используя предэкспериментальную ковариату, реализуем метод CUPED с помощью линейной регрессии:

def perform_ols_regression_cuped(data: pd.DataFrame):
    """Выполняет регрессию, включая ковариату baseline_metric в модель
    и возвращает информацию по модели.

    data - DataFrame с данными эксперимента

    return - таблица коэффициентов модели.
    """
    simple_model = smf.ols("y_post_metric ~ D + baseline_metric", 
                           data=data).fit()

    return simple_model.summary().tables[1]

simple_result = perform_ols_regression_cuped(data)
print(simple_result)

Стандартная ошибка стала ещё ниже, теперь она равна 0.035.

Используя все имеющиеся ковариаты, задействуем метод CUMPED через линейную регрессию:

def perform_ols_regression_multi_cuped(data: pd.DataFrame):
    """Выполняет Multi-CUPED с несколькими
    ковариатами (baseline_metric, covariate1, covariate2, covariate3).
    """
    multi_model = smf.ols(
        "y_post_metric ~ D + baseline_metric + covariate1 
         + covariate2 + covariate3",
        data=data
    ).fit()

    return multi_model.summary().tables[1]

multi_result = perform_ols_regression_multi_cuped(data)
print(multi_result)

Получили стандартную ошибку, равную 0.028, что ещё меньше, чем в предыдущих примерах.

Наконец объединим все методы вместе:

def perform_ols_all_methods(data: pd.DataFrame):
    """Выполняет ргерессию со
    всеми ковариатами (baseline_metric, strata,
    covariate1, covariate2, covariate3)."""
    multi_model = smf.ols(
        "y_post_metric ~ D + baseline_metric + strata 
      + covariate1 + covariate2 + covariate3",
        data=data
    ).fit()

    return multi_model.summary().tables[1]

all_result = perform_ols_all_methods(data)
print(all_result)

Как мы видим, таким образом, мы получили минимальное значение стандартной ошибки оценки коэффициента при D (0.012) по сравнению с другими методами по отдельности.

Общий вывод

Итак, линейная регрессия, как оказывается, является гораздо более универсальным инструментом, чем может показаться на первый взгляд. В этой статье мы продемонстрировали, что результаты многих статистических методов, применяемые в A/B тестировании, такие как t-тест, стратификация, CUPED, CUMPED в случае рандомизированного эксперимента можно получить путем построения линейной регрессии и проверки гипотезы в рамках построенной модели.

Получается, что все эти "новые" методы снижения дисперсии не такие уж и новые. Их всех можно представить через построение линейной регрессии, где главный подход сводится к довольно простой идее: добавление “хороших” переменных помогает уменьшить остаточную вариацию и улучшить точность оценки эффектов. 

Для более глубокого понимания этих методов и их теоретической основы рекомендуем обратиться к книге Causal Inference for The Brave and True и учебнику Ф. С. Картаева.

Над статьёй работали члены команды Ad-Hoc X5 Tech: Данила Яблочников и Денис Лавров.

Если эта статья заинтересовала вас, и вы хотели бы присоединиться к нашей команде, то мы с радостью приглашаем вас к нам! Отправьте своё резюме в Telegram @amSakhnov с пометкой «Статья». Мы будем рады познакомиться с вами и обсудить возможности сотрудничества.